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 DISPID propput_dispid
= DISPID_PROPERTYPUT
;
40 dynamic_var_t
*dynamic_vars
;
52 typedef HRESULT (*instr_func_t
)(exec_ctx_t
*);
82 static BOOL
lookup_dynamic_vars(dynamic_var_t
*var
, const WCHAR
*name
, ref_t
*ref
)
85 if(!wcsicmp(var
->name
, name
)) {
86 ref
->type
= var
->is_const
? REF_CONST
: REF_VAR
;
97 static BOOL
lookup_global_vars(ScriptDisp
*script
, const WCHAR
*name
, ref_t
*ref
)
99 dynamic_var_t
**vars
= script
->global_vars
;
100 size_t i
, cnt
= script
->global_vars_cnt
;
102 for(i
= 0; i
< cnt
; i
++) {
103 if(!wcsicmp(vars
[i
]->name
, name
)) {
104 ref
->type
= vars
[i
]->is_const
? REF_CONST
: REF_VAR
;
105 ref
->u
.v
= &vars
[i
]->v
;
113 static BOOL
lookup_global_funcs(ScriptDisp
*script
, const WCHAR
*name
, ref_t
*ref
)
115 function_t
**funcs
= script
->global_funcs
;
116 size_t i
, cnt
= script
->global_funcs_cnt
;
118 for(i
= 0; i
< cnt
; i
++) {
119 if(!wcsicmp(funcs
[i
]->name
, name
)) {
120 ref
->type
= REF_FUNC
;
129 static HRESULT
lookup_identifier(exec_ctx_t
*ctx
, BSTR name
, vbdisp_invoke_type_t invoke_type
, ref_t
*ref
)
131 ScriptDisp
*script_obj
= ctx
->script
->script_obj
;
137 if((ctx
->func
->type
== FUNC_FUNCTION
|| ctx
->func
->type
== FUNC_PROPGET
|| ctx
->func
->type
== FUNC_DEFGET
)
138 && !wcsicmp(name
, ctx
->func
->name
)) {
140 ref
->u
.v
= &ctx
->ret_val
;
144 if(ctx
->func
->type
!= FUNC_GLOBAL
) {
145 for(i
=0; i
< ctx
->func
->var_cnt
; i
++) {
146 if(!wcsicmp(ctx
->func
->vars
[i
].name
, name
)) {
148 ref
->u
.v
= ctx
->vars
+i
;
153 for(i
=0; i
< ctx
->func
->arg_cnt
; i
++) {
154 if(!wcsicmp(ctx
->func
->args
[i
].name
, name
)) {
156 ref
->u
.v
= ctx
->args
+i
;
161 if(lookup_dynamic_vars(ctx
->dynamic_vars
, name
, ref
))
165 /* FIXME: Bind such identifier while generating bytecode. */
166 for(i
=0; i
< ctx
->vbthis
->desc
->prop_cnt
; i
++) {
167 if(!wcsicmp(ctx
->vbthis
->desc
->props
[i
].name
, name
)) {
169 ref
->u
.v
= ctx
->vbthis
->props
+i
;
174 hres
= vbdisp_get_id(ctx
->vbthis
, name
, invoke_type
, TRUE
, &id
);
175 if(SUCCEEDED(hres
)) {
176 ref
->type
= REF_DISP
;
177 ref
->u
.d
.disp
= (IDispatch
*)&ctx
->vbthis
->IDispatchEx_iface
;
184 if(ctx
->code
->named_item
) {
185 if(lookup_global_vars(ctx
->code
->named_item
->script_obj
, name
, ref
))
187 if(lookup_global_funcs(ctx
->code
->named_item
->script_obj
, name
, ref
))
191 if(ctx
->func
->code_ctx
->named_item
&& ctx
->func
->code_ctx
->named_item
->disp
&&
192 !(ctx
->func
->code_ctx
->named_item
->flags
& SCRIPTITEM_CODEONLY
))
194 hres
= disp_get_id(ctx
->func
->code_ctx
->named_item
->disp
, name
, invoke_type
, TRUE
, &id
);
195 if(SUCCEEDED(hres
)) {
196 ref
->type
= REF_DISP
;
197 ref
->u
.d
.disp
= ctx
->func
->code_ctx
->named_item
->disp
;
203 if(lookup_global_vars(script_obj
, name
, ref
))
205 if(lookup_global_funcs(script_obj
, name
, ref
))
208 hres
= get_builtin_id(ctx
->script
->global_obj
, name
, &id
);
209 if(SUCCEEDED(hres
)) {
210 ref
->type
= REF_DISP
;
211 ref
->u
.d
.disp
= &ctx
->script
->global_obj
->IDispatch_iface
;
216 item
= lookup_named_item(ctx
->script
, name
, SCRIPTITEM_ISVISIBLE
);
217 if(item
&& item
->disp
) {
219 ref
->u
.obj
= item
->disp
;
223 LIST_FOR_EACH_ENTRY(item
, &ctx
->script
->named_items
, named_item_t
, entry
) {
224 if((item
->flags
& SCRIPTITEM_GLOBALMEMBERS
)) {
225 hres
= disp_get_id(item
->disp
, name
, invoke_type
, FALSE
, &id
);
226 if(SUCCEEDED(hres
)) {
227 ref
->type
= REF_DISP
;
228 ref
->u
.d
.disp
= item
->disp
;
235 ref
->type
= REF_NONE
;
239 static HRESULT
add_dynamic_var(exec_ctx_t
*ctx
, const WCHAR
*name
,
240 BOOL is_const
, VARIANT
**out_var
)
242 ScriptDisp
*script_obj
= ctx
->code
->named_item
? ctx
->code
->named_item
->script_obj
: ctx
->script
->script_obj
;
243 dynamic_var_t
*new_var
;
248 heap
= ctx
->func
->type
== FUNC_GLOBAL
? &script_obj
->heap
: &ctx
->heap
;
250 new_var
= heap_pool_alloc(heap
, sizeof(*new_var
));
252 return E_OUTOFMEMORY
;
254 size
= (lstrlenW(name
)+1)*sizeof(WCHAR
);
255 str
= heap_pool_alloc(heap
, size
);
257 return E_OUTOFMEMORY
;
258 memcpy(str
, name
, size
);
260 new_var
->is_const
= is_const
;
261 new_var
->array
= NULL
;
262 V_VT(&new_var
->v
) = VT_EMPTY
;
264 if(ctx
->func
->type
== FUNC_GLOBAL
) {
265 size_t cnt
= script_obj
->global_vars_cnt
+ 1;
266 if(cnt
> script_obj
->global_vars_size
) {
267 dynamic_var_t
**new_vars
;
268 if(script_obj
->global_vars
)
269 new_vars
= heap_realloc(script_obj
->global_vars
, cnt
* 2 * sizeof(*new_vars
));
271 new_vars
= heap_alloc(cnt
* 2 * sizeof(*new_vars
));
273 return E_OUTOFMEMORY
;
274 script_obj
->global_vars
= new_vars
;
275 script_obj
->global_vars_size
= cnt
* 2;
277 script_obj
->global_vars
[script_obj
->global_vars_cnt
++] = new_var
;
279 new_var
->next
= ctx
->dynamic_vars
;
280 ctx
->dynamic_vars
= new_var
;
283 *out_var
= &new_var
->v
;
287 void clear_ei(EXCEPINFO
*ei
)
289 SysFreeString(ei
->bstrSource
);
290 SysFreeString(ei
->bstrDescription
);
291 SysFreeString(ei
->bstrHelpFile
);
292 memset(ei
, 0, sizeof(*ei
));
295 static void clear_error_loc(script_ctx_t
*ctx
)
297 if(ctx
->error_loc_code
) {
298 release_vbscode(ctx
->error_loc_code
);
299 ctx
->error_loc_code
= NULL
;
303 static inline VARIANT
*stack_pop(exec_ctx_t
*ctx
)
306 return ctx
->stack
+ --ctx
->top
;
309 static inline VARIANT
*stack_top(exec_ctx_t
*ctx
, unsigned n
)
311 assert(ctx
->top
>= n
);
312 return ctx
->stack
+ (ctx
->top
-n
-1);
315 static HRESULT
stack_push(exec_ctx_t
*ctx
, VARIANT
*v
)
317 if(ctx
->stack_size
== ctx
->top
) {
320 new_stack
= heap_realloc(ctx
->stack
, ctx
->stack_size
*2*sizeof(*ctx
->stack
));
323 return E_OUTOFMEMORY
;
326 ctx
->stack
= new_stack
;
327 ctx
->stack_size
*= 2;
330 ctx
->stack
[ctx
->top
++] = *v
;
334 static inline HRESULT
stack_push_null(exec_ctx_t
*ctx
)
338 return stack_push(ctx
, &v
);
341 static void stack_popn(exec_ctx_t
*ctx
, unsigned n
)
344 VariantClear(stack_pop(ctx
));
347 static void stack_pop_deref(exec_ctx_t
*ctx
, variant_val_t
*r
)
352 if(V_VT(v
) == (VT_BYREF
|VT_VARIANT
)) {
354 r
->v
= V_VARIANTREF(v
);
361 static inline void release_val(variant_val_t
*v
)
367 static HRESULT
stack_pop_val(exec_ctx_t
*ctx
, variant_val_t
*r
)
369 stack_pop_deref(ctx
, r
);
371 if(V_VT(r
->v
) == VT_DISPATCH
) {
374 hres
= get_disp_value(ctx
->script
, V_DISPATCH(r
->v
), &r
->store
);
375 if(r
->owned
&& V_DISPATCH(r
->v
))
376 IDispatch_Release(V_DISPATCH(r
->v
));
387 static HRESULT
stack_assume_val(exec_ctx_t
*ctx
, unsigned n
)
389 VARIANT
*v
= stack_top(ctx
, n
);
392 if(V_VT(v
) == (VT_BYREF
|VT_VARIANT
)) {
393 VARIANT
*ref
= V_VARIANTREF(v
);
396 hres
= VariantCopy(v
, ref
);
401 if(V_VT(v
) == VT_DISPATCH
) {
404 disp
= V_DISPATCH(v
);
405 hres
= get_disp_value(ctx
->script
, disp
, v
);
407 IDispatch_Release(disp
);
415 static int stack_pop_bool(exec_ctx_t
*ctx
, BOOL
*b
)
420 hres
= stack_pop_val(ctx
, &val
);
440 FIXME("unsupported for %s\n", debugstr_variant(val
.v
));
447 static HRESULT
stack_pop_disp(exec_ctx_t
*ctx
, IDispatch
**ret
)
449 VARIANT
*v
= stack_pop(ctx
);
451 if(V_VT(v
) == VT_DISPATCH
) {
452 *ret
= V_DISPATCH(v
);
456 if(V_VT(v
) != (VT_VARIANT
|VT_BYREF
)) {
457 FIXME("not supported type: %s\n", debugstr_variant(v
));
463 if(V_VT(v
) != VT_DISPATCH
) {
464 FIXME("not disp %s\n", debugstr_variant(v
));
469 IDispatch_AddRef(V_DISPATCH(v
));
470 *ret
= V_DISPATCH(v
);
474 static HRESULT
stack_assume_disp(exec_ctx_t
*ctx
, unsigned n
, IDispatch
**disp
)
476 VARIANT
*v
= stack_top(ctx
, n
), *ref
;
478 if(V_VT(v
) != VT_DISPATCH
&& (disp
|| V_VT(v
) != VT_UNKNOWN
)) {
479 if(V_VT(v
) != (VT_VARIANT
|VT_BYREF
)) {
480 FIXME("not supported type: %s\n", debugstr_variant(v
));
484 ref
= V_VARIANTREF(v
);
485 if(V_VT(ref
) != VT_DISPATCH
&& (disp
|| V_VT(ref
) != VT_UNKNOWN
)) {
486 FIXME("not disp %s\n", debugstr_variant(ref
));
491 V_UNKNOWN(v
) = V_UNKNOWN(ref
);
493 IUnknown_AddRef(V_UNKNOWN(v
));
497 *disp
= V_DISPATCH(v
);
501 static inline void instr_jmp(exec_ctx_t
*ctx
, unsigned addr
)
503 ctx
->instr
= ctx
->code
->instrs
+ addr
;
506 static void vbstack_to_dp(exec_ctx_t
*ctx
, unsigned arg_cnt
, BOOL is_propput
, DISPPARAMS
*dp
)
508 dp
->cNamedArgs
= is_propput
? 1 : 0;
509 dp
->cArgs
= arg_cnt
+ dp
->cNamedArgs
;
510 dp
->rgdispidNamedArgs
= is_propput
? &propput_dispid
: NULL
;
516 assert(ctx
->top
>= arg_cnt
);
518 for(i
=1; i
*2 <= arg_cnt
; i
++) {
519 tmp
= ctx
->stack
[ctx
->top
-i
];
520 ctx
->stack
[ctx
->top
-i
] = ctx
->stack
[ctx
->top
-arg_cnt
+i
-1];
521 ctx
->stack
[ctx
->top
-arg_cnt
+i
-1] = tmp
;
524 dp
->rgvarg
= ctx
->stack
+ ctx
->top
-dp
->cArgs
;
526 dp
->rgvarg
= is_propput
? ctx
->stack
+ctx
->top
-1 : NULL
;
530 static HRESULT
array_access(exec_ctx_t
*ctx
, SAFEARRAY
*array
, DISPPARAMS
*dp
, VARIANT
**ret
)
532 unsigned i
, argc
= arg_cnt(dp
);
537 FIXME("NULL array\n");
541 hres
= SafeArrayLock(array
);
545 if(array
->cDims
!= argc
) {
546 FIXME("argc %d does not match cDims %d\n", dp
->cArgs
, array
->cDims
);
547 SafeArrayUnlock(array
);
551 indices
= heap_alloc(sizeof(*indices
) * argc
);
553 SafeArrayUnlock(array
);
554 return E_OUTOFMEMORY
;
557 for(i
=0; i
<argc
; i
++) {
558 hres
= to_int(get_arg(dp
, i
), indices
+i
);
561 SafeArrayUnlock(array
);
566 hres
= SafeArrayPtrOfIndex(array
, indices
, (void**)ret
);
567 SafeArrayUnlock(array
);
572 static HRESULT
variant_call(exec_ctx_t
*ctx
, VARIANT
*v
, unsigned arg_cnt
, VARIANT
*res
)
574 SAFEARRAY
*array
= NULL
;
578 TRACE("%s\n", debugstr_variant(v
));
580 if(V_VT(v
) == (VT_VARIANT
|VT_BYREF
))
584 case VT_ARRAY
|VT_BYREF
|VT_VARIANT
:
585 array
= *V_ARRAYREF(v
);
587 case VT_ARRAY
|VT_VARIANT
:
591 vbstack_to_dp(ctx
, arg_cnt
, FALSE
, &dp
);
592 hres
= disp_call(ctx
->script
, V_DISPATCH(v
), DISPID_VALUE
, &dp
, res
);
595 FIXME("unsupported on %s\n", debugstr_variant(v
));
605 vbstack_to_dp(ctx
, arg_cnt
, FALSE
, &dp
);
606 hres
= array_access(ctx
, array
, &dp
, &v
);
610 V_VT(res
) = VT_BYREF
|VT_VARIANT
;
614 stack_popn(ctx
, arg_cnt
);
618 static HRESULT
do_icall(exec_ctx_t
*ctx
, VARIANT
*res
)
620 BSTR identifier
= ctx
->instr
->arg1
.bstr
;
621 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
626 TRACE("%s %u\n", debugstr_w(identifier
), arg_cnt
);
628 hres
= lookup_identifier(ctx
, identifier
, VBDISP_CALLGET
, &ref
);
636 return variant_call(ctx
, ref
.u
.v
, arg_cnt
, res
);
639 FIXME("REF_VAR no res\n");
643 V_VT(res
) = VT_BYREF
|VT_VARIANT
;
644 V_BYREF(res
) = V_VT(ref
.u
.v
) == (VT_VARIANT
|VT_BYREF
) ? V_VARIANTREF(ref
.u
.v
) : ref
.u
.v
;
647 vbstack_to_dp(ctx
, arg_cnt
, FALSE
, &dp
);
648 hres
= disp_call(ctx
->script
, ref
.u
.d
.disp
, ref
.u
.d
.id
, &dp
, res
);
653 vbstack_to_dp(ctx
, arg_cnt
, FALSE
, &dp
);
654 hres
= exec_script(ctx
->script
, FALSE
, ref
.u
.f
, NULL
, &dp
, res
);
660 FIXME("arguments on object\n");
665 IDispatch_AddRef(ref
.u
.obj
);
666 V_VT(res
) = VT_DISPATCH
;
667 V_DISPATCH(res
) = ref
.u
.obj
;
671 if(res
&& !ctx
->func
->code_ctx
->option_explicit
&& arg_cnt
== 0) {
673 hres
= add_dynamic_var(ctx
, identifier
, FALSE
, &new);
676 V_VT(res
) = VT_BYREF
|VT_VARIANT
;
680 FIXME("%s not found\n", debugstr_w(identifier
));
681 return DISP_E_UNKNOWNNAME
;
684 stack_popn(ctx
, arg_cnt
);
688 static HRESULT
interp_icall(exec_ctx_t
*ctx
)
695 hres
= do_icall(ctx
, &v
);
699 return stack_push(ctx
, &v
);
702 static HRESULT
interp_icallv(exec_ctx_t
*ctx
)
705 return do_icall(ctx
, NULL
);
708 static HRESULT
interp_vcall(exec_ctx_t
*ctx
)
710 const unsigned arg_cnt
= ctx
->instr
->arg1
.uint
;
717 hres
= variant_call(ctx
, v
, arg_cnt
, &res
);
722 return stack_push(ctx
, &res
);
725 static HRESULT
interp_vcallv(exec_ctx_t
*ctx
)
727 const unsigned arg_cnt
= ctx
->instr
->arg1
.uint
;
734 hres
= variant_call(ctx
, v
, arg_cnt
, NULL
);
739 static HRESULT
do_mcall(exec_ctx_t
*ctx
, VARIANT
*res
)
741 const BSTR identifier
= ctx
->instr
->arg1
.bstr
;
742 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
748 hres
= stack_pop_disp(ctx
, &obj
);
757 vbstack_to_dp(ctx
, arg_cnt
, FALSE
, &dp
);
759 hres
= disp_get_id(obj
, identifier
, VBDISP_CALLGET
, FALSE
, &id
);
761 hres
= disp_call(ctx
->script
, obj
, id
, &dp
, res
);
762 IDispatch_Release(obj
);
766 stack_popn(ctx
, arg_cnt
);
770 static HRESULT
interp_mcall(exec_ctx_t
*ctx
)
777 hres
= do_mcall(ctx
, &res
);
781 return stack_push(ctx
, &res
);
784 static HRESULT
interp_mcallv(exec_ctx_t
*ctx
)
788 return do_mcall(ctx
, NULL
);
791 static HRESULT
assign_value(exec_ctx_t
*ctx
, VARIANT
*dst
, VARIANT
*src
, WORD flags
)
796 V_VT(&value
) = VT_EMPTY
;
797 hres
= VariantCopyInd(&value
, src
);
801 if(V_VT(&value
) == VT_DISPATCH
&& !(flags
& DISPATCH_PROPERTYPUTREF
)) {
802 IDispatch
*disp
= V_DISPATCH(&value
);
804 V_VT(&value
) = VT_EMPTY
;
805 hres
= get_disp_value(ctx
->script
, disp
, &value
);
807 IDispatch_Release(disp
);
817 static HRESULT
assign_ident(exec_ctx_t
*ctx
, BSTR name
, WORD flags
, DISPPARAMS
*dp
)
822 hres
= lookup_identifier(ctx
, name
, VBDISP_LET
, &ref
);
828 VARIANT
*v
= ref
.u
.v
;
830 if(V_VT(v
) == (VT_VARIANT
|VT_BYREF
))
836 if(V_VT(v
) == VT_DISPATCH
) {
837 hres
= disp_propput(ctx
->script
, V_DISPATCH(v
), DISPID_VALUE
, flags
, dp
);
841 if(!(V_VT(v
) & VT_ARRAY
)) {
842 FIXME("array assign on type %d\n", V_VT(v
));
847 case VT_ARRAY
|VT_BYREF
|VT_VARIANT
:
848 array
= *V_ARRAYREF(v
);
850 case VT_ARRAY
|VT_VARIANT
:
854 FIXME("Unsupported array type %x\n", V_VT(v
));
859 FIXME("null array\n");
863 hres
= array_access(ctx
, array
, dp
, &v
);
866 }else if(V_VT(v
) == (VT_ARRAY
|VT_BYREF
|VT_VARIANT
)) {
867 FIXME("non-array assign\n");
871 hres
= assign_value(ctx
, v
, dp
->rgvarg
, flags
);
875 hres
= disp_propput(ctx
->script
, ref
.u
.d
.disp
, ref
.u
.d
.id
, flags
, dp
);
878 FIXME("functions not implemented\n");
884 FIXME("REF_CONST\n");
887 if(ctx
->func
->code_ctx
->option_explicit
) {
888 FIXME("throw exception\n");
894 FIXME("arg_cnt %d not supported\n", arg_cnt(dp
));
898 TRACE("creating variable %s\n", debugstr_w(name
));
899 hres
= add_dynamic_var(ctx
, name
, FALSE
, &new_var
);
901 hres
= assign_value(ctx
, new_var
, dp
->rgvarg
, flags
);
908 static HRESULT
interp_assign_ident(exec_ctx_t
*ctx
)
910 const BSTR arg
= ctx
->instr
->arg1
.bstr
;
911 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
915 TRACE("%s\n", debugstr_w(arg
));
917 vbstack_to_dp(ctx
, arg_cnt
, TRUE
, &dp
);
918 hres
= assign_ident(ctx
, arg
, DISPATCH_PROPERTYPUT
, &dp
);
922 stack_popn(ctx
, arg_cnt
+1);
926 static HRESULT
interp_set_ident(exec_ctx_t
*ctx
)
928 const BSTR arg
= ctx
->instr
->arg1
.bstr
;
929 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
933 TRACE("%s %u\n", debugstr_w(arg
), arg_cnt
);
935 hres
= stack_assume_disp(ctx
, arg_cnt
, NULL
);
939 vbstack_to_dp(ctx
, arg_cnt
, TRUE
, &dp
);
940 hres
= assign_ident(ctx
, arg
, DISPATCH_PROPERTYPUTREF
, &dp
);
944 stack_popn(ctx
, arg_cnt
+ 1);
948 static HRESULT
interp_assign_member(exec_ctx_t
*ctx
)
950 BSTR identifier
= ctx
->instr
->arg1
.bstr
;
951 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
957 TRACE("%s\n", debugstr_w(identifier
));
959 hres
= stack_assume_disp(ctx
, arg_cnt
+1, &obj
);
968 hres
= disp_get_id(obj
, identifier
, VBDISP_LET
, FALSE
, &id
);
969 if(SUCCEEDED(hres
)) {
970 vbstack_to_dp(ctx
, arg_cnt
, TRUE
, &dp
);
971 hres
= disp_propput(ctx
->script
, obj
, id
, DISPATCH_PROPERTYPUT
, &dp
);
976 stack_popn(ctx
, arg_cnt
+2);
980 static HRESULT
interp_set_member(exec_ctx_t
*ctx
)
982 BSTR identifier
= ctx
->instr
->arg1
.bstr
;
983 const unsigned arg_cnt
= ctx
->instr
->arg2
.uint
;
989 TRACE("%s\n", debugstr_w(identifier
));
992 FIXME("arguments not supported\n");
996 hres
= stack_assume_disp(ctx
, 1, &obj
);
1001 FIXME("NULL obj\n");
1005 hres
= stack_assume_disp(ctx
, 0, NULL
);
1009 hres
= disp_get_id(obj
, identifier
, VBDISP_SET
, FALSE
, &id
);
1010 if(SUCCEEDED(hres
)) {
1011 vbstack_to_dp(ctx
, arg_cnt
, TRUE
, &dp
);
1012 hres
= disp_propput(ctx
->script
, obj
, id
, DISPATCH_PROPERTYPUTREF
, &dp
);
1021 static HRESULT
interp_const(exec_ctx_t
*ctx
)
1023 BSTR arg
= ctx
->instr
->arg1
.bstr
;
1028 TRACE("%s\n", debugstr_w(arg
));
1030 assert(ctx
->func
->type
== FUNC_GLOBAL
);
1032 hres
= lookup_identifier(ctx
, arg
, VBDISP_CALLGET
, &ref
);
1036 if(ref
.type
!= REF_NONE
) {
1037 FIXME("%s already defined\n", debugstr_w(arg
));
1041 hres
= stack_assume_val(ctx
, 0);
1045 hres
= add_dynamic_var(ctx
, arg
, TRUE
, &v
);
1049 *v
= *stack_pop(ctx
);
1053 static HRESULT
interp_val(exec_ctx_t
*ctx
)
1061 hres
= stack_pop_val(ctx
, &val
);
1066 V_VT(&v
) = VT_EMPTY
;
1067 hres
= VariantCopy(&v
, val
.v
);
1072 return stack_push(ctx
, val
.owned
? val
.v
: &v
);
1075 static HRESULT
interp_pop(exec_ctx_t
*ctx
)
1077 const unsigned n
= ctx
->instr
->arg1
.uint
;
1085 static HRESULT
interp_stack(exec_ctx_t
*ctx
)
1087 const unsigned n
= ctx
->instr
->arg1
.uint
;
1094 return MAKE_VBSERROR(505);
1095 assert(n
< ctx
->top
);
1097 V_VT(&v
) = VT_EMPTY
;
1098 hres
= VariantCopy(&v
, ctx
->stack
+ n
);
1102 return stack_push(ctx
, &v
);
1105 static HRESULT
interp_deref(exec_ctx_t
*ctx
)
1107 VARIANT copy
, *v
= stack_top(ctx
, 0);
1110 TRACE("%s\n", debugstr_variant(v
));
1112 if(V_VT(v
) != (VT_BYREF
|VT_VARIANT
))
1115 V_VT(©
) = VT_EMPTY
;
1116 hres
= VariantCopy(©
, V_VARIANTREF(v
));
1122 static HRESULT
interp_new(exec_ctx_t
*ctx
)
1124 const WCHAR
*arg
= ctx
->instr
->arg1
.bstr
;
1125 class_desc_t
*class_desc
= NULL
;
1130 static const WCHAR regexpW
[] = {'r','e','g','e','x','p',0};
1132 TRACE("%s\n", debugstr_w(arg
));
1134 if(!wcsicmp(arg
, regexpW
)) {
1135 V_VT(&v
) = VT_DISPATCH
;
1136 hres
= create_regexp(&V_DISPATCH(&v
));
1140 return stack_push(ctx
, &v
);
1143 if(ctx
->code
->named_item
)
1144 for(class_desc
= ctx
->code
->named_item
->script_obj
->classes
; class_desc
; class_desc
= class_desc
->next
)
1145 if(!wcsicmp(class_desc
->name
, arg
))
1148 for(class_desc
= ctx
->script
->script_obj
->classes
; class_desc
; class_desc
= class_desc
->next
)
1149 if(!wcsicmp(class_desc
->name
, arg
))
1152 FIXME("Class %s not found\n", debugstr_w(arg
));
1156 hres
= create_vbdisp(class_desc
, &obj
);
1160 V_VT(&v
) = VT_DISPATCH
;
1161 V_DISPATCH(&v
) = (IDispatch
*)&obj
->IDispatchEx_iface
;
1162 return stack_push(ctx
, &v
);
1165 static HRESULT
interp_dim(exec_ctx_t
*ctx
)
1167 ScriptDisp
*script_obj
= ctx
->code
->named_item
? ctx
->code
->named_item
->script_obj
: ctx
->script
->script_obj
;
1168 const BSTR ident
= ctx
->instr
->arg1
.bstr
;
1169 const unsigned array_id
= ctx
->instr
->arg2
.uint
;
1170 const array_desc_t
*array_desc
;
1171 SAFEARRAY
**array_ref
;
1175 TRACE("%s\n", debugstr_w(ident
));
1177 assert(array_id
< ctx
->func
->array_cnt
);
1179 if(ctx
->func
->type
== FUNC_GLOBAL
) {
1181 for(i
= 0; i
< script_obj
->global_vars_cnt
; i
++) {
1182 if(!wcsicmp(script_obj
->global_vars
[i
]->name
, ident
))
1185 assert(i
< script_obj
->global_vars_cnt
);
1186 v
= &script_obj
->global_vars
[i
]->v
;
1187 array_ref
= &script_obj
->global_vars
[i
]->array
;
1192 ctx
->arrays
= heap_alloc_zero(ctx
->func
->array_cnt
* sizeof(SAFEARRAY
*));
1194 return E_OUTOFMEMORY
;
1197 hres
= lookup_identifier(ctx
, ident
, VBDISP_LET
, &ref
);
1199 FIXME("lookup %s failed: %08x\n", debugstr_w(ident
), hres
);
1203 if(ref
.type
!= REF_VAR
) {
1204 FIXME("got ref.type = %d\n", ref
.type
);
1209 array_ref
= ctx
->arrays
+ array_id
;
1213 FIXME("Array already initialized\n");
1217 array_desc
= ctx
->func
->array_descs
+ array_id
;
1218 if(array_desc
->dim_cnt
) {
1219 *array_ref
= SafeArrayCreate(VT_VARIANT
, array_desc
->dim_cnt
, array_desc
->bounds
);
1221 return E_OUTOFMEMORY
;
1224 V_VT(v
) = VT_ARRAY
|VT_BYREF
|VT_VARIANT
;
1225 V_ARRAYREF(v
) = array_ref
;
1229 static HRESULT
array_bounds_from_stack(exec_ctx_t
*ctx
, unsigned dim_cnt
, SAFEARRAYBOUND
**ret
)
1231 SAFEARRAYBOUND
*bounds
;
1236 if(!(bounds
= heap_alloc(dim_cnt
* sizeof(*bounds
))))
1237 return E_OUTOFMEMORY
;
1239 for(i
= 0; i
< dim_cnt
; i
++) {
1240 hres
= to_int(stack_top(ctx
, dim_cnt
- i
- 1), &dim
);
1246 bounds
[i
].cElements
= dim
+ 1;
1247 bounds
[i
].lLbound
= 0;
1250 stack_popn(ctx
, dim_cnt
);
1255 static HRESULT
interp_redim(exec_ctx_t
*ctx
)
1257 BSTR identifier
= ctx
->instr
->arg1
.bstr
;
1258 const unsigned dim_cnt
= ctx
->instr
->arg2
.uint
;
1259 SAFEARRAYBOUND
*bounds
;
1264 TRACE("%s %u\n", debugstr_w(identifier
), dim_cnt
);
1266 hres
= lookup_identifier(ctx
, identifier
, VBDISP_LET
, &ref
);
1268 FIXME("lookup %s failed: %08x\n", debugstr_w(identifier
), hres
);
1272 if(ref
.type
!= REF_VAR
) {
1273 FIXME("got ref.type = %d\n", ref
.type
);
1277 hres
= array_bounds_from_stack(ctx
, dim_cnt
, &bounds
);
1281 array
= SafeArrayCreate(VT_VARIANT
, dim_cnt
, bounds
);
1284 return E_OUTOFMEMORY
;
1286 /* FIXME: We should check if we're not modifying an existing static array here */
1288 VariantClear(ref
.u
.v
);
1289 V_VT(ref
.u
.v
) = VT_ARRAY
|VT_VARIANT
;
1290 V_ARRAY(ref
.u
.v
) = array
;
1294 static HRESULT
interp_step(exec_ctx_t
*ctx
)
1296 const BSTR ident
= ctx
->instr
->arg2
.bstr
;
1302 TRACE("%s\n", debugstr_w(ident
));
1304 V_VT(&zero
) = VT_I2
;
1306 hres
= VarCmp(stack_top(ctx
, 0), &zero
, ctx
->script
->lcid
, 0);
1310 gteq_zero
= hres
== VARCMP_GT
|| hres
== VARCMP_EQ
;
1312 hres
= lookup_identifier(ctx
, ident
, VBDISP_ANY
, &ref
);
1316 if(ref
.type
!= REF_VAR
) {
1317 FIXME("%s is not REF_VAR\n", debugstr_w(ident
));
1321 hres
= VarCmp(ref
.u
.v
, stack_top(ctx
, 1), ctx
->script
->lcid
, 0);
1325 if(hres
== VARCMP_EQ
|| hres
== (gteq_zero
? VARCMP_LT
: VARCMP_GT
)) {
1329 instr_jmp(ctx
, ctx
->instr
->arg1
.uint
);
1334 static HRESULT
interp_newenum(exec_ctx_t
*ctx
)
1342 stack_pop_deref(ctx
, &v
);
1343 assert(V_VT(stack_top(ctx
, 0)) == VT_EMPTY
);
1344 r
= stack_top(ctx
, 0);
1347 case VT_DISPATCH
|VT_BYREF
:
1350 DISPPARAMS dp
= {0};
1353 hres
= disp_call(ctx
->script
, V_ISBYREF(v
.v
) ? *V_DISPATCHREF(v
.v
) : V_DISPATCH(v
.v
), DISPID_NEWENUM
, &dp
, &iterv
);
1358 if(V_VT(&iterv
) != VT_UNKNOWN
&& V_VT(&iterv
) != VT_DISPATCH
) {
1359 FIXME("Unsupported iterv %s\n", debugstr_variant(&iterv
));
1360 VariantClear(&iterv
);
1364 hres
= IUnknown_QueryInterface(V_UNKNOWN(&iterv
), &IID_IEnumVARIANT
, (void**)&iter
);
1365 IUnknown_Release(V_UNKNOWN(&iterv
));
1367 FIXME("Could not get IEnumVARIANT iface: %08x\n", hres
);
1371 V_VT(r
) = VT_UNKNOWN
;
1372 V_UNKNOWN(r
) = (IUnknown
*)iter
;
1375 case VT_VARIANT
|VT_ARRAY
:
1376 case VT_VARIANT
|VT_ARRAY
|VT_BYREF
: {
1379 hres
= create_safearray_iter(V_ISBYREF(v
.v
) ? *V_ARRAYREF(v
.v
) : V_ARRAY(v
.v
), &iter
);
1383 V_VT(r
) = VT_UNKNOWN
;
1384 V_UNKNOWN(r
) = (IUnknown
*)iter
;
1388 FIXME("Unsupported for %s\n", debugstr_variant(v
.v
));
1396 static HRESULT
interp_enumnext(exec_ctx_t
*ctx
)
1398 const unsigned loop_end
= ctx
->instr
->arg1
.uint
;
1399 const BSTR ident
= ctx
->instr
->arg2
.bstr
;
1401 DISPPARAMS dp
= {&v
, &propput_dispid
, 1, 1};
1408 if(V_VT(stack_top(ctx
, 0)) == VT_EMPTY
) {
1409 FIXME("uninitialized\n");
1413 assert(V_VT(stack_top(ctx
, 0)) == VT_UNKNOWN
);
1414 iter
= (IEnumVARIANT
*)V_UNKNOWN(stack_top(ctx
, 0));
1416 V_VT(&v
) = VT_EMPTY
;
1417 hres
= IEnumVARIANT_Next(iter
, 1, &v
, NULL
);
1421 do_continue
= hres
== S_OK
;
1422 hres
= assign_ident(ctx
, ident
, DISPATCH_PROPERTYPUT
|DISPATCH_PROPERTYPUTREF
, &dp
);
1431 instr_jmp(ctx
, loop_end
);
1436 static HRESULT
interp_jmp(exec_ctx_t
*ctx
)
1438 const unsigned arg
= ctx
->instr
->arg1
.uint
;
1442 instr_jmp(ctx
, arg
);
1446 static HRESULT
interp_jmp_false(exec_ctx_t
*ctx
)
1448 const unsigned arg
= ctx
->instr
->arg1
.uint
;
1454 hres
= stack_pop_bool(ctx
, &b
);
1461 instr_jmp(ctx
, ctx
->instr
->arg1
.uint
);
1465 static HRESULT
interp_jmp_true(exec_ctx_t
*ctx
)
1467 const unsigned arg
= ctx
->instr
->arg1
.uint
;
1473 hres
= stack_pop_bool(ctx
, &b
);
1478 instr_jmp(ctx
, ctx
->instr
->arg1
.uint
);
1484 static HRESULT
interp_ret(exec_ctx_t
*ctx
)
1492 static HRESULT
interp_retval(exec_ctx_t
*ctx
)
1499 stack_pop_deref(ctx
, &val
);
1502 VariantClear(&ctx
->ret_val
);
1503 ctx
->ret_val
= *val
.v
;
1506 hres
= VariantCopy(&ctx
->ret_val
, val
.v
);
1514 static HRESULT
interp_stop(exec_ctx_t
*ctx
)
1518 /* NOTE: this should have effect in debugging mode (that we don't support yet) */
1522 static HRESULT
interp_me(exec_ctx_t
*ctx
)
1530 disp
= (IDispatch
*)&ctx
->vbthis
->IDispatchEx_iface
;
1531 }else if(ctx
->code
->named_item
) {
1532 disp
= (ctx
->code
->named_item
->flags
& SCRIPTITEM_CODEONLY
)
1533 ? (IDispatch
*)&ctx
->code
->named_item
->script_obj
->IDispatchEx_iface
1534 : ctx
->code
->named_item
->disp
;
1538 LIST_FOR_EACH_ENTRY(item
, &ctx
->script
->named_items
, named_item_t
, entry
) {
1539 if(!(item
->flags
& SCRIPTITEM_GLOBALMEMBERS
)) continue;
1544 disp
= (IDispatch
*)&ctx
->script
->script_obj
->IDispatchEx_iface
;
1547 IDispatch_AddRef(disp
);
1548 V_VT(&v
) = VT_DISPATCH
;
1549 V_DISPATCH(&v
) = disp
;
1550 return stack_push(ctx
, &v
);
1553 static HRESULT
interp_bool(exec_ctx_t
*ctx
)
1555 const VARIANT_BOOL arg
= ctx
->instr
->arg1
.lng
;
1558 TRACE("%s\n", arg
? "true" : "false");
1562 return stack_push(ctx
, &v
);
1565 static HRESULT
interp_errmode(exec_ctx_t
*ctx
)
1567 const int err_mode
= ctx
->instr
->arg1
.uint
;
1569 TRACE("%d\n", err_mode
);
1571 ctx
->resume_next
= err_mode
;
1572 clear_ei(&ctx
->script
->ei
);
1576 static HRESULT
interp_string(exec_ctx_t
*ctx
)
1583 V_BSTR(&v
) = SysAllocString(ctx
->instr
->arg1
.str
);
1585 return E_OUTOFMEMORY
;
1587 return stack_push(ctx
, &v
);
1590 static HRESULT
interp_int(exec_ctx_t
*ctx
)
1592 const LONG arg
= ctx
->instr
->arg1
.lng
;
1597 if(arg
== (INT16
)arg
) {
1604 return stack_push(ctx
, &v
);
1607 static HRESULT
interp_double(exec_ctx_t
*ctx
)
1609 const DOUBLE
*arg
= ctx
->instr
->arg1
.dbl
;
1612 TRACE("%lf\n", *arg
);
1616 return stack_push(ctx
, &v
);
1619 static HRESULT
interp_empty(exec_ctx_t
*ctx
)
1625 V_VT(&v
) = VT_EMPTY
;
1626 return stack_push(ctx
, &v
);
1629 static HRESULT
interp_null(exec_ctx_t
*ctx
)
1632 return stack_push_null(ctx
);
1635 static HRESULT
interp_nothing(exec_ctx_t
*ctx
)
1641 V_VT(&v
) = VT_DISPATCH
;
1642 V_DISPATCH(&v
) = NULL
;
1643 return stack_push(ctx
, &v
);
1646 static HRESULT
interp_hres(exec_ctx_t
*ctx
)
1648 const unsigned arg
= ctx
->instr
->arg1
.uint
;
1653 V_VT(&v
) = VT_ERROR
;
1655 return stack_push(ctx
, &v
);
1658 static HRESULT
interp_not(exec_ctx_t
*ctx
)
1666 hres
= stack_pop_val(ctx
, &val
);
1670 hres
= VarNot(val
.v
, &v
);
1675 return stack_push(ctx
, &v
);
1678 static HRESULT
interp_and(exec_ctx_t
*ctx
)
1686 hres
= stack_pop_val(ctx
, &r
);
1690 hres
= stack_pop_val(ctx
, &l
);
1691 if(SUCCEEDED(hres
)) {
1692 hres
= VarAnd(l
.v
, r
.v
, &v
);
1699 return stack_push(ctx
, &v
);
1702 static HRESULT
interp_or(exec_ctx_t
*ctx
)
1710 hres
= stack_pop_val(ctx
, &r
);
1714 hres
= stack_pop_val(ctx
, &l
);
1715 if(SUCCEEDED(hres
)) {
1716 hres
= VarOr(l
.v
, r
.v
, &v
);
1723 return stack_push(ctx
, &v
);
1726 static HRESULT
interp_xor(exec_ctx_t
*ctx
)
1734 hres
= stack_pop_val(ctx
, &r
);
1738 hres
= stack_pop_val(ctx
, &l
);
1739 if(SUCCEEDED(hres
)) {
1740 hres
= VarXor(l
.v
, r
.v
, &v
);
1747 return stack_push(ctx
, &v
);
1750 static HRESULT
interp_eqv(exec_ctx_t
*ctx
)
1758 hres
= stack_pop_val(ctx
, &r
);
1762 hres
= stack_pop_val(ctx
, &l
);
1763 if(SUCCEEDED(hres
)) {
1764 hres
= VarEqv(l
.v
, r
.v
, &v
);
1771 return stack_push(ctx
, &v
);
1774 static HRESULT
interp_imp(exec_ctx_t
*ctx
)
1782 hres
= stack_pop_val(ctx
, &r
);
1786 hres
= stack_pop_val(ctx
, &l
);
1787 if(SUCCEEDED(hres
)) {
1788 hres
= VarImp(l
.v
, r
.v
, &v
);
1795 return stack_push(ctx
, &v
);
1798 static HRESULT
var_cmp(exec_ctx_t
*ctx
, VARIANT
*l
, VARIANT
*r
)
1800 TRACE("%s %s\n", debugstr_variant(l
), debugstr_variant(r
));
1802 /* FIXME: Fix comparing string to number */
1804 return VarCmp(l
, r
, ctx
->script
->lcid
, 0);
1807 static HRESULT
cmp_oper(exec_ctx_t
*ctx
)
1812 hres
= stack_pop_val(ctx
, &r
);
1816 hres
= stack_pop_val(ctx
, &l
);
1817 if(SUCCEEDED(hres
)) {
1818 hres
= var_cmp(ctx
, l
.v
, r
.v
);
1826 static HRESULT
interp_equal(exec_ctx_t
*ctx
)
1833 hres
= cmp_oper(ctx
);
1836 if(hres
== VARCMP_NULL
)
1837 return stack_push_null(ctx
);
1840 V_BOOL(&v
) = hres
== VARCMP_EQ
? VARIANT_TRUE
: VARIANT_FALSE
;
1841 return stack_push(ctx
, &v
);
1844 static HRESULT
interp_nequal(exec_ctx_t
*ctx
)
1851 hres
= cmp_oper(ctx
);
1854 if(hres
== VARCMP_NULL
)
1855 return stack_push_null(ctx
);
1858 V_BOOL(&v
) = hres
!= VARCMP_EQ
? VARIANT_TRUE
: VARIANT_FALSE
;
1859 return stack_push(ctx
, &v
);
1862 static HRESULT
interp_gt(exec_ctx_t
*ctx
)
1869 hres
= cmp_oper(ctx
);
1872 if(hres
== VARCMP_NULL
)
1873 return stack_push_null(ctx
);
1876 V_BOOL(&v
) = hres
== VARCMP_GT
? VARIANT_TRUE
: VARIANT_FALSE
;
1877 return stack_push(ctx
, &v
);
1880 static HRESULT
interp_gteq(exec_ctx_t
*ctx
)
1887 hres
= cmp_oper(ctx
);
1890 if(hres
== VARCMP_NULL
)
1891 return stack_push_null(ctx
);
1894 V_BOOL(&v
) = hres
== VARCMP_GT
|| hres
== VARCMP_EQ
? VARIANT_TRUE
: VARIANT_FALSE
;
1895 return stack_push(ctx
, &v
);
1898 static HRESULT
interp_lt(exec_ctx_t
*ctx
)
1905 hres
= cmp_oper(ctx
);
1908 if(hres
== VARCMP_NULL
)
1909 return stack_push_null(ctx
);
1912 V_BOOL(&v
) = hres
== VARCMP_LT
? VARIANT_TRUE
: VARIANT_FALSE
;
1913 return stack_push(ctx
, &v
);
1916 static HRESULT
interp_lteq(exec_ctx_t
*ctx
)
1923 hres
= cmp_oper(ctx
);
1926 if(hres
== VARCMP_NULL
)
1927 return stack_push_null(ctx
);
1930 V_BOOL(&v
) = hres
== VARCMP_LT
|| hres
== VARCMP_EQ
? VARIANT_TRUE
: VARIANT_FALSE
;
1931 return stack_push(ctx
, &v
);
1934 static HRESULT
interp_case(exec_ctx_t
*ctx
)
1936 const unsigned arg
= ctx
->instr
->arg1
.uint
;
1942 hres
= stack_pop_val(ctx
, &v
);
1946 hres
= var_cmp(ctx
, stack_top(ctx
, 0), v
.v
);
1951 if(hres
== VARCMP_EQ
) {
1953 instr_jmp(ctx
, arg
);
1961 static HRESULT
interp_is(exec_ctx_t
*ctx
)
1963 IUnknown
*l
= NULL
, *r
= NULL
;
1965 HRESULT hres
= S_OK
;
1969 stack_pop_deref(ctx
, &v
);
1970 if(V_VT(v
.v
) != VT_DISPATCH
&& V_VT(v
.v
) != VT_UNKNOWN
) {
1971 FIXME("Unhandled type %s\n", debugstr_variant(v
.v
));
1973 }else if(V_UNKNOWN(v
.v
)) {
1974 hres
= IUnknown_QueryInterface(V_UNKNOWN(v
.v
), &IID_IUnknown
, (void**)&r
);
1976 if(v
.owned
) VariantClear(v
.v
);
1980 stack_pop_deref(ctx
, &v
);
1981 if(V_VT(v
.v
) != VT_DISPATCH
&& V_VT(v
.v
) != VT_UNKNOWN
) {
1982 FIXME("Unhandled type %s\n", debugstr_variant(v
.v
));
1984 }else if(V_UNKNOWN(v
.v
)) {
1985 hres
= IUnknown_QueryInterface(V_UNKNOWN(v
.v
), &IID_IUnknown
, (void**)&l
);
1987 if(v
.owned
) VariantClear(v
.v
);
1989 if(SUCCEEDED(hres
)) {
1991 V_VT(&res
) = VT_BOOL
;
1993 V_BOOL(&res
) = VARIANT_TRUE
;
1995 V_BOOL(&res
) = VARIANT_FALSE
;
1997 IObjectIdentity
*identity
;
1998 hres
= IUnknown_QueryInterface(l
, &IID_IObjectIdentity
, (void**)&identity
);
1999 if(SUCCEEDED(hres
)) {
2000 hres
= IObjectIdentity_IsEqualObject(identity
, r
);
2001 IObjectIdentity_Release(identity
);
2003 V_BOOL(&res
) = hres
== S_OK
? VARIANT_TRUE
: VARIANT_FALSE
;
2005 hres
= stack_push(ctx
, &res
);
2008 IUnknown_Release(r
);
2010 IUnknown_Release(l
);
2014 static HRESULT
interp_concat(exec_ctx_t
*ctx
)
2022 hres
= stack_pop_val(ctx
, &r
);
2026 hres
= stack_pop_val(ctx
, &l
);
2027 if(SUCCEEDED(hres
)) {
2028 hres
= VarCat(l
.v
, r
.v
, &v
);
2035 return stack_push(ctx
, &v
);
2038 static HRESULT
interp_add(exec_ctx_t
*ctx
)
2046 hres
= stack_pop_val(ctx
, &r
);
2050 hres
= stack_pop_val(ctx
, &l
);
2051 if(SUCCEEDED(hres
)) {
2052 hres
= VarAdd(l
.v
, r
.v
, &v
);
2059 return stack_push(ctx
, &v
);
2062 static HRESULT
interp_sub(exec_ctx_t
*ctx
)
2070 hres
= stack_pop_val(ctx
, &r
);
2074 hres
= stack_pop_val(ctx
, &l
);
2075 if(SUCCEEDED(hres
)) {
2076 hres
= VarSub(l
.v
, r
.v
, &v
);
2083 return stack_push(ctx
, &v
);
2086 static HRESULT
interp_mod(exec_ctx_t
*ctx
)
2094 hres
= stack_pop_val(ctx
, &r
);
2098 hres
= stack_pop_val(ctx
, &l
);
2099 if(SUCCEEDED(hres
)) {
2100 hres
= VarMod(l
.v
, r
.v
, &v
);
2107 return stack_push(ctx
, &v
);
2110 static HRESULT
interp_idiv(exec_ctx_t
*ctx
)
2118 hres
= stack_pop_val(ctx
, &r
);
2122 hres
= stack_pop_val(ctx
, &l
);
2123 if(SUCCEEDED(hres
)) {
2124 hres
= VarIdiv(l
.v
, r
.v
, &v
);
2131 return stack_push(ctx
, &v
);
2134 static HRESULT
interp_div(exec_ctx_t
*ctx
)
2142 hres
= stack_pop_val(ctx
, &r
);
2146 hres
= stack_pop_val(ctx
, &l
);
2147 if(SUCCEEDED(hres
)) {
2148 hres
= VarDiv(l
.v
, r
.v
, &v
);
2155 return stack_push(ctx
, &v
);
2158 static HRESULT
interp_mul(exec_ctx_t
*ctx
)
2166 hres
= stack_pop_val(ctx
, &r
);
2170 hres
= stack_pop_val(ctx
, &l
);
2171 if(SUCCEEDED(hres
)) {
2172 hres
= VarMul(l
.v
, r
.v
, &v
);
2179 return stack_push(ctx
, &v
);
2182 static HRESULT
interp_exp(exec_ctx_t
*ctx
)
2190 hres
= stack_pop_val(ctx
, &r
);
2194 hres
= stack_pop_val(ctx
, &l
);
2195 if(SUCCEEDED(hres
)) {
2196 hres
= VarPow(l
.v
, r
.v
, &v
);
2203 return stack_push(ctx
, &v
);
2206 static HRESULT
interp_neg(exec_ctx_t
*ctx
)
2212 hres
= stack_pop_val(ctx
, &val
);
2216 hres
= VarNeg(val
.v
, &v
);
2221 return stack_push(ctx
, &v
);
2224 static HRESULT
interp_incc(exec_ctx_t
*ctx
)
2226 const BSTR ident
= ctx
->instr
->arg1
.bstr
;
2233 hres
= lookup_identifier(ctx
, ident
, VBDISP_LET
, &ref
);
2237 if(ref
.type
!= REF_VAR
) {
2238 FIXME("ref.type is not REF_VAR\n");
2242 hres
= VarAdd(stack_top(ctx
, 0), ref
.u
.v
, &v
);
2246 VariantClear(ref
.u
.v
);
2251 static HRESULT
interp_catch(exec_ctx_t
*ctx
)
2253 /* Nothing to do here, the OP is for unwinding only. */
2257 static const instr_func_t op_funcs
[] = {
2258 #define X(x,n,a,b) interp_ ## x,
2263 static const unsigned op_move
[] = {
2264 #define X(x,n,a,b) n,
2269 void release_dynamic_var(dynamic_var_t
*var
)
2271 VariantClear(&var
->v
);
2273 SafeArrayDestroy(var
->array
);
2276 static void release_exec(exec_ctx_t
*ctx
)
2281 VariantClear(&ctx
->ret_val
);
2283 for(var
= ctx
->dynamic_vars
; var
; var
= var
->next
)
2284 release_dynamic_var(var
);
2287 IDispatchEx_Release(&ctx
->vbthis
->IDispatchEx_iface
);
2290 for(i
=0; i
< ctx
->func
->arg_cnt
; i
++)
2291 VariantClear(ctx
->args
+i
);
2295 for(i
=0; i
< ctx
->func
->var_cnt
; i
++)
2296 VariantClear(ctx
->vars
+i
);
2300 for(i
=0; i
< ctx
->func
->array_cnt
; i
++) {
2302 SafeArrayDestroy(ctx
->arrays
[i
]);
2304 heap_free(ctx
->arrays
);
2307 heap_pool_free(&ctx
->heap
);
2308 heap_free(ctx
->args
);
2309 heap_free(ctx
->vars
);
2310 heap_free(ctx
->stack
);
2313 HRESULT
exec_script(script_ctx_t
*ctx
, BOOL extern_caller
, function_t
*func
, vbdisp_t
*vbthis
, DISPPARAMS
*dp
, VARIANT
*res
)
2315 exec_ctx_t exec
= {func
->code_ctx
};
2317 HRESULT hres
= S_OK
;
2319 exec
.code
= func
->code_ctx
;
2321 if(dp
? func
->arg_cnt
!= arg_cnt(dp
) : func
->arg_cnt
) {
2322 FIXME("wrong arg_cnt %d, expected %d\n", dp
? arg_cnt(dp
) : 0, func
->arg_cnt
);
2326 heap_pool_init(&exec
.heap
);
2328 TRACE("%s(", debugstr_w(func
->name
));
2333 exec
.args
= heap_alloc_zero(func
->arg_cnt
* sizeof(VARIANT
));
2335 release_exec(&exec
);
2336 return E_OUTOFMEMORY
;
2339 for(i
=0; i
< func
->arg_cnt
; i
++) {
2341 TRACE("%s%s", i
? ", " : "", debugstr_variant(v
));
2342 if(V_VT(v
) == (VT_VARIANT
|VT_BYREF
)) {
2343 if(func
->args
[i
].by_ref
)
2346 hres
= VariantCopyInd(exec
.args
+i
, V_VARIANTREF(v
));
2348 hres
= VariantCopyInd(exec
.args
+i
, v
);
2351 release_exec(&exec
);
2361 exec
.vars
= heap_alloc_zero(func
->var_cnt
* sizeof(VARIANT
));
2363 release_exec(&exec
);
2364 return E_OUTOFMEMORY
;
2370 exec
.stack_size
= 16;
2372 exec
.stack
= heap_alloc(exec
.stack_size
* sizeof(VARIANT
));
2374 release_exec(&exec
);
2375 return E_OUTOFMEMORY
;
2379 IActiveScriptSite_OnEnterScript(ctx
->site
);
2382 IDispatchEx_AddRef(&vbthis
->IDispatchEx_iface
);
2383 exec
.vbthis
= vbthis
;
2386 exec
.instr
= exec
.code
->instrs
+ func
->code_off
;
2391 op
= exec
.instr
->op
;
2392 hres
= op_funcs
[op
](&exec
);
2394 if(hres
!= SCRIPT_E_RECORDED
) {
2397 ctx
->ei
.scode
= hres
= map_hres(hres
);
2398 ctx
->ei
.bstrSource
= get_vbscript_string(VBS_RUNTIME_ERROR
);
2399 ctx
->ei
.bstrDescription
= get_vbscript_error_string(hres
);
2401 hres
= ctx
->ei
.scode
;
2404 if(exec
.resume_next
) {
2407 WARN("Failed %08x in resume next mode\n", hres
);
2410 * Unwinding here is simple. We need to find the next OP_catch, which contains
2411 * information about expected stack size and jump offset on error. Generated
2412 * bytecode needs to guarantee, that simple jump and stack adjustment will
2413 * guarantee proper execution continuation.
2415 while((++exec
.instr
)->op
!= OP_catch
);
2417 TRACE("unwind jmp %d stack_off %d\n", exec
.instr
->arg1
.uint
, exec
.instr
->arg2
.uint
);
2419 clear_error_loc(ctx
);
2420 stack_off
= exec
.instr
->arg2
.uint
;
2421 instr_jmp(&exec
, exec
.instr
->arg1
.uint
);
2423 if(exec
.top
> stack_off
) {
2424 stack_popn(&exec
, exec
.top
-stack_off
);
2425 }else if(exec
.top
< stack_off
) {
2428 V_VT(&v
) = VT_EMPTY
;
2429 while(exec
.top
< stack_off
) {
2430 hres
= stack_push(&exec
, &v
);
2438 if(!ctx
->error_loc_code
) {
2439 grab_vbscode(exec
.code
);
2440 ctx
->error_loc_code
= exec
.code
;
2441 ctx
->error_loc_offset
= exec
.instr
->loc
;
2443 stack_popn(&exec
, exec
.top
);
2448 exec
.instr
+= op_move
[op
];
2456 ctx
->ei
.scode
= hres
;
2457 hres
= report_script_error(ctx
, ctx
->error_loc_code
, ctx
->error_loc_offset
);
2458 clear_error_loc(ctx
);
2460 IActiveScriptSite_OnLeaveScript(ctx
->site
);
2463 if(SUCCEEDED(hres
) && res
) {
2464 *res
= exec
.ret_val
;
2465 V_VT(&exec
.ret_val
) = VT_EMPTY
;
2468 release_exec(&exec
);