patch 8.2.0657: Vim9: no check if called variable is a FuncRef
Problem: Vim9: no check if called variable is a FuncRef.
Solution: Add a type check.
diff --git a/src/vim9compile.c b/src/vim9compile.c
index 89a8677..aa3cd07 100644
--- a/src/vim9compile.c
+++ b/src/vim9compile.c
@@ -1327,15 +1327,32 @@
/*
* Generate an ISN_PCALL instruction.
+ * "type" is the type of the FuncRef.
*/
static int
-generate_PCALL(cctx_T *cctx, int argcount, int at_top)
+generate_PCALL(
+ cctx_T *cctx,
+ int argcount,
+ char_u *name,
+ type_T *type,
+ int at_top)
{
isn_T *isn;
garray_T *stack = &cctx->ctx_type_stack;
+ type_T *ret_type;
RETURN_OK_IF_SKIP(cctx);
+ if (type->tt_type == VAR_ANY)
+ ret_type = &t_any;
+ else if (type->tt_type == VAR_FUNC || type->tt_type == VAR_PARTIAL)
+ ret_type = type->tt_member;
+ else
+ {
+ semsg(_("E1085: Not a callable type: %s"), name);
+ return FAIL;
+ }
+
if ((isn = generate_instr(cctx, ISN_PCALL)) == NULL)
return FAIL;
isn->isn_arg.pfunc.cpf_top = at_top;
@@ -1344,7 +1361,7 @@
stack->ga_len -= argcount; // drop the arguments
// drop the funcref/partial, get back the return value
- ((type_T **)stack->ga_data)[stack->ga_len - 1] = &t_any;
+ ((type_T **)stack->ga_data)[stack->ga_len - 1] = ret_type;
// If partial is above the arguments it must be cleared and replaced with
// the return value.
@@ -2465,12 +2482,16 @@
if (STRNCMP(namebuf, "g:", 2) != 0
&& compile_load(&p, namebuf + varlen, cctx, FALSE) == OK)
{
- res = generate_PCALL(cctx, argcount, FALSE);
+ garray_T *stack = &cctx->ctx_type_stack;
+ type_T *type;
+
+ type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
+ res = generate_PCALL(cctx, argcount, namebuf, type, FALSE);
goto theend;
}
// A global function may be defined only later. Need to figure out at
- // runtime.
+ // runtime. Also handles a FuncRef at runtime.
if (STRNCMP(namebuf, "g:", 2) == 0)
res = generate_UCALL(cctx, name, argcount);
else
@@ -3120,13 +3141,17 @@
{
if (**arg == '(')
{
- int argcount = 0;
+ garray_T *stack = &cctx->ctx_type_stack;
+ type_T *type;
+ int argcount = 0;
// funcref(arg)
+ type = ((type_T **)stack->ga_data)[stack->ga_len - 1];
+
*arg = skipwhite(*arg + 1);
if (compile_arguments(arg, cctx, &argcount) == FAIL)
return FAIL;
- if (generate_PCALL(cctx, argcount, TRUE) == FAIL)
+ if (generate_PCALL(cctx, argcount, end_leader, type, TRUE) == FAIL)
return FAIL;
}
else if (**arg == '-' && (*arg)[1] == '>')