From: Bram Moolenaar Date: Sun, 31 Jan 2021 16:48:30 +0000 (+0100) Subject: patch 8.2.2441: Vim9: extend() does not give an error for a type mismatch X-Git-Tag: v8.2.2441 X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=c03f5c677a1fba99d2379550ccf2391eaa43e2ac;p=vim patch 8.2.2441: Vim9: extend() does not give an error for a type mismatch Problem: Vim9: extend() does not give an error for a type mismatch. Solution: Check the type of the second argument. (closes #7760) --- diff --git a/src/list.c b/src/list.c index 77dd2a307..ca6e469c4 100644 --- a/src/list.c +++ b/src/list.c @@ -2493,6 +2493,16 @@ f_count(typval_T *argvars, typval_T *rettv) static void extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) { + type_T *type = NULL; + garray_T type_list; + + if (!is_new && in_vim9script()) + { + // Check that map() does not change the type of the dict. + ga_init2(&type_list, sizeof(type_T *), 10); + type = typval2type(argvars, &type_list); + } + if (argvars[0].v_type == VAR_LIST && argvars[1].v_type == VAR_LIST) { list_T *l1, *l2; @@ -2504,7 +2514,7 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) if (l1 == NULL) { emsg(_(e_cannot_extend_null_list)); - return; + goto theend; } l2 = argvars[1].vval.v_list; if ((is_new || !value_check_lock(l1->lv_lock, arg_errmsg, TRUE)) @@ -2514,14 +2524,14 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) { l1 = list_copy(l1, FALSE, get_copyID()); if (l1 == NULL) - return; + goto theend; } if (argvars[2].v_type != VAR_UNKNOWN) { before = (long)tv_get_number_chk(&argvars[2], &error); if (error) - return; // type error; errmsg already given + goto theend; // type error; errmsg already given if (before == l1->lv_len) item = NULL; @@ -2531,12 +2541,14 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) if (item == NULL) { semsg(_(e_listidx), before); - return; + goto theend; } } } else item = NULL; + if (type != NULL && check_typval_type(type, &argvars[1], 2) == FAIL) + goto theend; list_extend(l1, l2, item); if (is_new) @@ -2559,7 +2571,7 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) if (d1 == NULL) { emsg(_(e_cannot_extend_null_dict)); - return; + goto theend; } d2 = argvars[1].vval.v_dict; if ((is_new || !value_check_lock(d1->dv_lock, arg_errmsg, TRUE)) @@ -2569,7 +2581,7 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) { d1 = dict_copy(d1, FALSE, get_copyID()); if (d1 == NULL) - return; + goto theend; } // Check the third argument. @@ -2579,19 +2591,21 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) action = tv_get_string_chk(&argvars[2]); if (action == NULL) - return; // type error; errmsg already given + goto theend; // type error; errmsg already given for (i = 0; i < 3; ++i) if (STRCMP(action, av[i]) == 0) break; if (i == 3) { semsg(_(e_invarg2), action); - return; + goto theend; } } else action = (char_u *)"force"; + if (type != NULL && check_typval_type(type, &argvars[1], 2) == FAIL) + goto theend; dict_extend(d1, d2, action); if (is_new) @@ -2606,6 +2620,10 @@ extend(typval_T *argvars, typval_T *rettv, char_u *arg_errmsg, int is_new) } else semsg(_(e_listdictarg), is_new ? "extendnew()" : "extend()"); + +theend: + if (type != NULL) + clear_type_list(&type_list); } /* diff --git a/src/testdir/test_vim9_builtin.vim b/src/testdir/test_vim9_builtin.vim index d9275e617..e7bec71c8 100644 --- a/src/testdir/test_vim9_builtin.vim +++ b/src/testdir/test_vim9_builtin.vim @@ -240,19 +240,24 @@ def Test_expand() enddef def Test_extend_arg_types() - assert_equal([1, 2, 3], extend([1, 2], [3])) - assert_equal([3, 1, 2], extend([1, 2], [3], 0)) - assert_equal([1, 3, 2], extend([1, 2], [3], 1)) - assert_equal([1, 3, 2], extend([1, 2], [3], s:number_one)) - - assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3})) - assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4})) - assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep')) - assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, s:string_keep)) - - var res: list> - extend(res, mapnew([1, 2], (_, v) => ({}))) - assert_equal([{}, {}], res) + g:number_one = 1 + g:string_keep = 'keep' + var lines =<< trim END + assert_equal([1, 2, 3], extend([1, 2], [3])) + assert_equal([3, 1, 2], extend([1, 2], [3], 0)) + assert_equal([1, 3, 2], extend([1, 2], [3], 1)) + assert_equal([1, 3, 2], extend([1, 2], [3], g:number_one)) + + assert_equal({a: 1, b: 2, c: 3}, extend({a: 1, b: 2}, {c: 3})) + assert_equal({a: 1, b: 4}, extend({a: 1, b: 2}, {b: 4})) + assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, 'keep')) + assert_equal({a: 1, b: 2}, extend({a: 1, b: 2}, {b: 4}, g:string_keep)) + + var res: list> + extend(res, mapnew([1, 2], (_, v) => ({}))) + assert_equal([{}, {}], res) + END + CheckDefAndScriptSuccess(lines) CheckDefFailure(['extend([1, 2], 3)'], 'E1013: Argument 2: type mismatch, expected list but got number') CheckDefFailure(['extend([1, 2], ["x"])'], 'E1013: Argument 2: type mismatch, expected list but got list') @@ -300,8 +305,7 @@ def Test_extend_dict_item_type() var d: dict = {a: 1} extend(d, {b: 'x'}) END - CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected dict but got dict', 2) - CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3) + CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected dict but got dict', 2) lines =<< trim END var d: dict = {a: 1} @@ -326,8 +330,7 @@ def Test_extend_list_item_type() var l: list = [1] extend(l, ['x']) END - CheckDefFailure(lines, 'E1013: Argument 2: type mismatch, expected list but got list', 2) - CheckScriptFailure(['vim9script'] + lines, 'E1012:', 3) + CheckDefAndScriptFailure(lines, 'E1013: Argument 2: type mismatch, expected list but got list', 2) lines =<< trim END var l: list = [1] diff --git a/src/version.c b/src/version.c index c5074c274..5555c7712 100644 --- a/src/version.c +++ b/src/version.c @@ -750,6 +750,8 @@ static char *(features[]) = static int included_patches[] = { /* Add new patch number below this line */ +/**/ + 2441, /**/ 2440, /**/