0
0
mirror of https://github.com/vim/vim.git synced 2025-09-24 03:44:06 -04:00

patch 8.2.3650: Vim9: for loop variable can be a list member

Problem:    Vim9: for loop variable can be a list member.
Solution:   Check for valid variable name. (closes #9179)
This commit is contained in:
Bram Moolenaar
2021-11-22 20:10:18 +00:00
parent 7a53f29c03
commit 3b3755fe19
7 changed files with 38 additions and 9 deletions

View File

@@ -1102,7 +1102,7 @@ dict_extend(dict_T *d1, dict_T *d2, char_u *action, char *func_name)
&& HI2DI(hi2)->di_tv.v_type == VAR_FUNC && HI2DI(hi2)->di_tv.v_type == VAR_FUNC
&& var_wrong_func_name(hi2->hi_key, di1 == NULL)) && var_wrong_func_name(hi2->hi_key, di1 == NULL))
break; break;
if (!valid_varname(hi2->hi_key, TRUE)) if (!valid_varname(hi2->hi_key, -1, TRUE))
break; break;
} }

View File

@@ -1128,7 +1128,7 @@ get_lval(
wrong = (lp->ll_dict->dv_scope == VAR_DEF_SCOPE wrong = (lp->ll_dict->dv_scope == VAR_DEF_SCOPE
&& rettv->v_type == VAR_FUNC && rettv->v_type == VAR_FUNC
&& var_wrong_func_name(key, lp->ll_di == NULL)) && var_wrong_func_name(key, lp->ll_di == NULL))
|| !valid_varname(key, TRUE); || !valid_varname(key, -1, TRUE);
if (len != -1) if (len != -1)
key[len] = prevval; key[len] = prevval;
if (wrong) if (wrong)

View File

@@ -3431,7 +3431,7 @@ set_var_const(
// Make sure the variable name is valid. In Vim9 script an autoload // Make sure the variable name is valid. In Vim9 script an autoload
// variable must be prefixed with "g:". // variable must be prefixed with "g:".
if (!valid_varname(varname, !vim9script if (!valid_varname(varname, -1, !vim9script
|| STRNCMP(name, "g:", 2) == 0)) || STRNCMP(name, "g:", 2) == 0))
goto failed; goto failed;
@@ -3631,14 +3631,15 @@ value_check_lock(int lock, char_u *name, int use_gettext)
/* /*
* Check if a variable name is valid. When "autoload" is true "#" is allowed. * Check if a variable name is valid. When "autoload" is true "#" is allowed.
* If "len" is -1 use all of "varname", otherwise up to "varname[len]".
* Return FALSE and give an error if not. * Return FALSE and give an error if not.
*/ */
int int
valid_varname(char_u *varname, int autoload) valid_varname(char_u *varname, int len, int autoload)
{ {
char_u *p; char_u *p;
for (p = varname; *p != NUL; ++p) for (p = varname; len < 0 ? *p != NUL : p < varname + len; ++p)
if (!eval_isnamec1(*p) && (p == varname || !VIM_ISDIGIT(*p)) if (!eval_isnamec1(*p) && (p == varname || !VIM_ISDIGIT(*p))
&& !(autoload && *p == AUTOLOAD_CHAR)) && !(autoload && *p == AUTOLOAD_CHAR))
{ {

View File

@@ -78,7 +78,7 @@ int var_check_lock(int flags, char_u *name, int use_gettext);
int var_check_fixed(int flags, char_u *name, int use_gettext); int var_check_fixed(int flags, char_u *name, int use_gettext);
int var_wrong_func_name(char_u *name, int new_var); int var_wrong_func_name(char_u *name, int new_var);
int value_check_lock(int lock, char_u *name, int use_gettext); int value_check_lock(int lock, char_u *name, int use_gettext);
int valid_varname(char_u *varname, int autoload); int valid_varname(char_u *varname, int len, int autoload);
void reset_v_option_vars(void); void reset_v_option_vars(void);
void assert_error(garray_T *gap); void assert_error(garray_T *gap);
int var_exists(char_u *var); int var_exists(char_u *var);

View File

@@ -2865,11 +2865,35 @@ def Test_for_loop_fails()
endfor endfor
END END
CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected job but got string', 2) CheckDefExecAndScriptFailure(lines, 'E1012: Type mismatch; expected job but got string', 2)
lines =<< trim END
var i = 0
for i in [1, 2, 3]
echo i
endfor
END
CheckDefExecAndScriptFailure2(lines, 'E1017:', 'E1041:')
lines =<< trim END
var l = [0]
for l[0] in [1, 2, 3]
echo l[0]
endfor
END
CheckDefExecAndScriptFailure2(lines, 'E461:', 'E1017:')
lines =<< trim END
var d = {x: 0}
for d.x in [1, 2, 3]
echo d.x
endfor
END
CheckDefExecAndScriptFailure2(lines, 'E461:', 'E1017:')
enddef enddef
def Test_for_loop_script_var() def Test_for_loop_script_var()
# cannot use s:var in a :def function # cannot use s:var in a :def function
CheckDefFailure(['for s:var in range(3)', 'echo 3'], 'E1101:') CheckDefFailure(['for s:var in range(3)', 'echo 3'], 'E461:')
# can use s:var in Vim9 script, with or without s: # can use s:var in Vim9 script, with or without s:
var lines =<< trim END var lines =<< trim END

View File

@@ -757,6 +757,8 @@ static char *(features[]) =
static int included_patches[] = static int included_patches[] =
{ /* Add new patch number below this line */ { /* Add new patch number below this line */
/**/
3650,
/**/ /**/
3649, 3649,
/**/ /**/

View File

@@ -8220,8 +8220,8 @@ compile_for(char_u *arg_start, cctx_T *cctx)
for (idx = 0; idx < var_count; ++idx) for (idx = 0; idx < var_count; ++idx)
{ {
assign_dest_T dest = dest_local; assign_dest_T dest = dest_local;
int opt_flags = 0; int opt_flags = 0;
int vimvaridx = -1; int vimvaridx = -1;
type_T *type = &t_any; type_T *type = &t_any;
type_T *lhs_type = &t_any; type_T *lhs_type = &t_any;
where_T where = WHERE_INIT; where_T where = WHERE_INIT;
@@ -8255,6 +8255,8 @@ compile_for(char_u *arg_start, cctx_T *cctx)
} }
else else
{ {
if (!valid_varname(arg, varlen, FALSE))
goto failed;
if (lookup_local(arg, varlen, NULL, cctx) == OK) if (lookup_local(arg, varlen, NULL, cctx) == OK)
{ {
semsg(_(e_variable_already_declared), arg); semsg(_(e_variable_already_declared), arg);