Skip to content

Commit

Permalink
patch 8.2.3091: Vim9: default argument expr. cannot use previous argu…
Browse files Browse the repository at this point in the history
…ment

Problem:    Vim9: default argument expression cannot use previous argument
Solution:   Correct argument index. (closes #8496)
  • Loading branch information
brammool committed Jul 3, 2021
1 parent 00aaa51 commit e28d9b3
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 4 deletions.
2 changes: 2 additions & 0 deletions src/structs.h
Expand Up @@ -1610,6 +1610,8 @@ typedef struct
int uf_dfunc_idx; // only valid if uf_def_status is UF_COMPILED
garray_T uf_args; // arguments, including optional arguments
garray_T uf_def_args; // default argument expressions
int uf_args_visible; // normally uf_args.ga_len, less when
// compiling default argument expression.

// for :def (for :function uf_ret_type is NULL)
type_T **uf_arg_types; // argument types (count == uf_args.ga_len)
Expand Down
6 changes: 6 additions & 0 deletions src/testdir/test_vim9_func.vim
Expand Up @@ -452,6 +452,12 @@ def Test_call_default_args()
MyDefaultThird('->', 'xx', v:none)->assert_equal('->xxbb')
MyDefaultThird('->', v:none, 'yy')->assert_equal('->aayy')
MyDefaultThird('->', 'xx', 'yy')->assert_equal('->xxyy')

def DefArg(mandatory: any, optional = mandatory): string
return mandatory .. optional
enddef
DefArg(1234)->assert_equal('12341234')
DefArg("ok")->assert_equal('okok')
END
CheckDefAndScriptSuccess(lines)

Expand Down
2 changes: 2 additions & 0 deletions src/version.c
Expand Up @@ -755,6 +755,8 @@ static char *(features[]) =

static int included_patches[] =
{ /* Add new patch number below this line */
/**/
3091,
/**/
3090,
/**/
Expand Down
7 changes: 3 additions & 4 deletions src/vim9compile.c
Expand Up @@ -274,7 +274,7 @@ arg_exists(

if (len == 0)
return FAIL;
for (idx = 0; idx < cctx->ctx_ufunc->uf_args.ga_len; ++idx)
for (idx = 0; idx < cctx->ctx_ufunc->uf_args_visible; ++idx)
{
char_u *arg = FUNCARG(cctx->ctx_ufunc, idx);

Expand Down Expand Up @@ -9172,7 +9172,6 @@ compile_def_function(
{
int count = ufunc->uf_def_args.ga_len;
int first_def_arg = ufunc->uf_args.ga_len - count;
int uf_args_len = ufunc->uf_args.ga_len;
int i;
char_u *arg;
int off = STACK_FRAME_SIZE + (ufunc->uf_va_name != NULL ? 1 : 0);
Expand All @@ -9195,12 +9194,11 @@ compile_def_function(
goto erret;

// Make sure later arguments are not found.
ufunc->uf_args.ga_len = i;
ufunc->uf_args_visible = arg_idx;

arg = ((char_u **)(ufunc->uf_def_args.ga_data))[i];
r = compile_expr0(&arg, &cctx);

ufunc->uf_args.ga_len = uf_args_len;
if (r == FAIL)
goto erret;

Expand Down Expand Up @@ -9230,6 +9228,7 @@ compile_def_function(
if (did_set_arg_type)
set_function_type(ufunc);
}
ufunc->uf_args_visible = ufunc->uf_args.ga_len;

/*
* Loop over all the lines of the function and generate instructions.
Expand Down

0 comments on commit e28d9b3

Please sign in to comment.