diff --git a/clang-tools-extra/pseudo/lib/GLR.cpp b/clang-tools-extra/pseudo/lib/GLR.cpp --- a/clang-tools-extra/pseudo/lib/GLR.cpp +++ b/clang-tools-extra/pseudo/lib/GLR.cpp @@ -182,9 +182,13 @@ for (const PlaceholderRecovery *Option : BestOptions) { const ForestNode &Placeholder = Params.Forest.createOpaque(Option->Symbol, RecoveryRange->Begin); - const GSS::Node *NewHead = Params.GSStack.addNode( - *Lang.Table.getGoToState(Option->RecoveryNode->State, Option->Symbol), - &Placeholder, {Option->RecoveryNode}); + LRTable::StateID OldState = Option->RecoveryNode->State; + LRTable::StateID NewState = + isToken(Option->Symbol) + ? *Lang.Table.getShiftState(OldState, Option->Symbol) + : *Lang.Table.getGoToState(OldState, Option->Symbol); + const GSS::Node *NewHead = + Params.GSStack.addNode(NewState, &Placeholder, {Option->RecoveryNode}); NewHeads.push_back(NewHead); } TokenIndex = RecoveryRange->End; diff --git a/clang-tools-extra/pseudo/unittests/GLRTest.cpp b/clang-tools-extra/pseudo/unittests/GLRTest.cpp --- a/clang-tools-extra/pseudo/unittests/GLRTest.cpp +++ b/clang-tools-extra/pseudo/unittests/GLRTest.cpp @@ -604,6 +604,28 @@ "[ 5, end) └─} := tok[5]\n"); } +TEST_F(GLRTest, RecoverTerminal) { + build(R"bnf( + _ := stmt + + stmt := IDENTIFIER ; [recover=Skip] + )bnf"); + TestLang.Table = LRTable::buildSLR(TestLang.G); + TestLang.RecoveryStrategies.try_emplace( + extensionID("Skip"), + [](Token::Index Start, const TokenStream &) { return Start + 1; }); + clang::LangOptions LOptions; + TokenStream Tokens = cook(lex("foo", LOptions), LOptions); + + const ForestNode &Parsed = + glrParse({Tokens, Arena, GSStack}, id("stmt"), TestLang); + EXPECT_EQ(Parsed.dumpRecursive(TestLang.G), + "[ 0, end) stmt := IDENTIFIER ; [recover=Skip]\n" + "[ 0, 1) ├─IDENTIFIER := tok[0]\n" + "[ 1, end) └─; := \n"); +} + + TEST_F(GLRTest, NoExplicitAccept) { build(R"bnf( _ := test