commit 1817ccdb107ceeaf5c48fe193da5146682c15ca6
author zeertzjq <zeertzjq@outlook.com> Wed Apr 10 17:37:47 2024 +0200
committer Christian Brabandt <cb@256bit.org> Wed Apr 10 17:37:47 2024 +0200
tree 1b0fecad67841b8cd509468fad8ee6a3d0957fdb
parent aa8e22b035dd669cb47c9237f9c9a917ec6a0ec4

patch 9.1.0301: Vim9: heredoc start may be recognized in string

Problem:  Vim9: heredoc start may be recognized in string.
Solution: Don't skip to closing bracket for invalid list assignment.
          (zeertzjq)

closes: #14472

Signed-off-by: zeertzjq <zeertzjq@outlook.com>
Signed-off-by: Christian Brabandt <cb@256bit.org>

src/testdir/test_let.vim

diff --git a/src/testdir/test_let.vim b/src/testdir/test_let.vim
index 1d61635..5ee2e9b 100644
--- a/src/testdir/test_let.vim
+++ b/src/testdir/test_let.vim
@@ -409,10 +409,16 @@
   endtry
 
   try
+    let [] =<< trim TEXT
+    TEXT
+    call assert_report('No exception thrown')
+  catch /E475:/
+  catch
+    call assert_report('Caught exception: ' .. v:exception)
+  endtry
+
+  try
     let [a b c] =<< trim TEXT
-      change
-      insert
-      append
     TEXT
     call assert_report('No exception thrown')
   catch /E475:/
@@ -422,9 +428,6 @@
 
   try
     let [a; b; c] =<< trim TEXT
-      change
-      insert
-      append
     TEXT
     call assert_report('No exception thrown')
   catch /E452:/

src/testdir/test_vim9_assign.vim

diff --git a/src/testdir/test_vim9_assign.vim b/src/testdir/test_vim9_assign.vim
index 0d6d783..0beaa44 100644
--- a/src/testdir/test_vim9_assign.vim
+++ b/src/testdir/test_vim9_assign.vim
@@ -2029,6 +2029,20 @@
   END
   v9.CheckScriptSuccess(lines)
 
+  # heredoc start should not be recognized in string
+  lines =<< trim END
+      vim9script
+      def Func()
+        new
+        @" = 'bar'
+        ['foo', @"]->setline("]=<<"->count('='))
+        assert_equal(['foo', 'bar'], getline(1, '$'))
+        bwipe!
+      enddef
+      Func()
+  END
+  v9.CheckScriptSuccess(lines)
+
   v9.CheckDefFailure(['var lines =<< trim END X', 'END'], 'E488:')
   v9.CheckDefFailure(['var lines =<< trim END " comment', 'END'], 'E488:')
 

src/userfunc.c

diff --git a/src/userfunc.c b/src/userfunc.c
index 015733c..71b3983 100644
--- a/src/userfunc.c
+++ b/src/userfunc.c
@@ -1230,18 +1230,15 @@
 		    int		save_sc_version = current_sctx.sc_version;
 		    int		var_count = 0;
 		    int		semicolon = 0;
-		    char_u	*argend;
 
 		    current_sctx.sc_version
 				     = vim9_function ? SCRIPT_VERSION_VIM9 : 1;
-		    argend = skip_var_list(arg, TRUE, &var_count, &semicolon,
+		    arg = skip_var_list(arg, TRUE, &var_count, &semicolon,
 									 TRUE);
-		    if (argend == NULL)
-			// Invalid list assignment: skip to closing bracket.
-			argend = find_name_end(arg, NULL, NULL, FNE_INCL_BR);
-		    arg = skipwhite(argend);
+		    if (arg != NULL)
+			arg = skipwhite(arg);
 		    current_sctx.sc_version = save_sc_version;
-		    if (arg[0] == '=' && arg[1] == '<' && arg[2] =='<')
+		    if (arg != NULL && STRNCMP(arg, "=<<", 3) == 0)
 		    {
 			p = skipwhite(arg + 3);
 			while (TRUE)

src/version.c

diff --git a/src/version.c b/src/version.c
index c256520..2f049c9 100644
--- a/src/version.c
+++ b/src/version.c
@@ -705,6 +705,8 @@
 static int included_patches[] =
 {   /* Add new patch number below this line */
 /**/
+    301,
+/**/
     300,
 /**/
     299,