diff options
author | Bram Moolenaar <Bram@vim.org> | 2021-01-30 23:05:11 +0100 |
---|---|---|
committer | Bram Moolenaar <Bram@vim.org> | 2021-01-30 23:05:11 +0100 |
commit | f2b26bcf8f498fed72759af4aa768fb2aab3118c (patch) | |
tree | 3fb840f0bd9de41b463443ddf17131a0ad77f226 /src/evalfunc.c | |
parent | b63f3ca66db20f7b135aae706d36be538f91f8cc (diff) |
patch 8.2.2434: Vim9: no error when compiling str2nr() with a numberv8.2.2434
Problem: Vim9: no error when compiling str2nr() with a number.
Solution: Add argument type checks. (closes #7759)
Diffstat (limited to 'src/evalfunc.c')
-rw-r--r-- | src/evalfunc.c | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/src/evalfunc.c b/src/evalfunc.c index b752c51d10..1fad082ac0 100644 --- a/src/evalfunc.c +++ b/src/evalfunc.c @@ -331,6 +331,18 @@ arg_string(type_T *type, argcontext_T *context) } /* + * Check "type" is a bool or number 0 or 1. + */ + static int +arg_bool(type_T *type, argcontext_T *context) +{ + if (type->tt_type == VAR_ANY + || type->tt_type == VAR_NUMBER || type->tt_type == VAR_BOOL) + return OK; + return check_arg_type(&t_bool, type, context); +} + +/* * Check "type" is a list or a blob. */ static int @@ -423,6 +435,7 @@ arg_extend3(type_T *type, argcontext_T *context) /* * Lists of functions that check the argument types of a builtin function. */ +argcheck_T arg3_string_nr_bool[] = {arg_string, arg_number, arg_bool}; argcheck_T arg1_float_or_nr[] = {arg_float_or_nr}; argcheck_T arg2_listblob_item[] = {arg_list_or_blob, arg_item_of_prev}; argcheck_T arg23_extend[] = {arg_list_or_dict, arg_same_as_prev, arg_extend3}; @@ -1552,7 +1565,7 @@ static funcentry_T global_functions[] = ret_float, FLOAT_FUNC(f_str2float)}, {"str2list", 1, 2, FEARG_1, NULL, ret_list_number, f_str2list}, - {"str2nr", 1, 3, FEARG_1, NULL, + {"str2nr", 1, 3, FEARG_1, arg3_string_nr_bool, ret_number, f_str2nr}, {"strcharpart", 2, 3, FEARG_1, NULL, ret_string, f_strcharpart}, @@ -9076,7 +9089,7 @@ f_str2nr(typval_T *argvars, typval_T *rettv) what |= STR2NR_QUOTE; } - p = skipwhite(tv_get_string(&argvars[0])); + p = skipwhite(tv_get_string_strict(&argvars[0])); isneg = (*p == '-'); if (*p == '+' || *p == '-') p = skipwhite(p + 1); |