[Compiler] Fix parse error when parsing for loops

Author: WGB5445

third_party/move/move-compiler/src/parser/lexer.rs

@@ -356,7 +356,7 @@ impl<'input> Lexer<'input> {
             let offset = self.text.len() - text.len();
             let (found_token, length) = find_token(self.file_hash, text, offset)?;
             token = found_token;
-            current_offset += length;
+            current_offset = offset + length;
         }
         Ok(token)
     }

third_party/move/move-compiler/src/parser/syntax.rs

@@ -947,8 +947,8 @@ fn parse_term(context: &mut Context) -> Result<Exp, Box<Diagnostic>> {
         },
         Tok::Identifier
             if context.tokens.content() == FOR_IDENT
-                && matches!(context.tokens.lookahead_nth(1), Ok(Tok::LParen))
-                && matches!(context.tokens.lookahead_nth(3), Ok(Tok::Identifier)) =>
+                && matches!(context.tokens.lookahead_nth(0), Ok(Tok::LParen))
+                && matches!(context.tokens.lookahead_nth(2), Ok(Tok::Identifier)) =>
         {
             let (control_exp, _) = parse_for_loop(context)?;
             // for loop isn't useful in an expression, so we ignore second result from

third_party/move/move-compiler/tests/move_check/control_flow/for_loop_comment_following.move

@@ -0,0 +1,9 @@
+//# run
+script {
+    fun main(): () {
+        for/**/( i in 0..10) {};
+        for/**/ ( i in 0..10) {};
+        for /**/( i in 0..10) {};
+        for /**/ ( i in 0..10) {};
+    }
+}

third_party/move/move-compiler/tests/move_check/control_flow/for_loop_lparen_following.move

@@ -0,0 +1,6 @@
+//# run
+script {
+    fun main(): () {
+        for( i in 0..10) {};
+    }
+}