diff --git a/clang-tools-extra/clangd/ConfigYAML.cpp b/clang-tools-extra/clangd/ConfigYAML.cpp --- a/clang-tools-extra/clangd/ConfigYAML.cpp +++ b/clang-tools-extra/clangd/ConfigYAML.cpp @@ -26,6 +26,7 @@ class Parser { llvm::SourceMgr &SM; + bool HadError = false; public: Parser(llvm::SourceMgr &SM) : SM(SM) {} @@ -35,40 +36,38 @@ // The private parse() helpers follow the same pattern. bool parse(Fragment &F, Node &N) { DictParser Dict("Config", this); - Dict.handle("If", [&](Node &N) { return parse(F.If, N); }); - Dict.handle("CompileFlags", - [&](Node &N) { return parse(F.CompileFlags, N); }); - return Dict.parse(N); + Dict.handle("If", [&](Node &N) { parse(F.If, N); }); + Dict.handle("CompileFlags", [&](Node &N) { parse(F.CompileFlags, N); }); + Dict.parse(N); + return !(N.failed() || HadError); } private: - bool parse(Fragment::IfBlock &F, Node &N) { + void parse(Fragment::IfBlock &F, Node &N) { DictParser Dict("If", this); Dict.unrecognized( [&](llvm::StringRef) { F.HasUnrecognizedCondition = true; }); Dict.handle("PathMatch", [&](Node &N) { if (auto Values = scalarValues(N)) F.PathMatch = std::move(*Values); - return !N.failed(); }); - return Dict.parse(N); + Dict.parse(N); } - bool parse(Fragment::CompileFlagsBlock &F, Node &N) { + void parse(Fragment::CompileFlagsBlock &F, Node &N) { DictParser Dict("CompileFlags", this); Dict.handle("Add", [&](Node &N) { if (auto Values = scalarValues(N)) F.Add = std::move(*Values); - return !N.failed(); }); - return Dict.parse(N); + Dict.parse(N); } // Helper for parsing mapping nodes (dictionaries). // We don't use YamlIO as we want to control over unknown keys. class DictParser { llvm::StringRef Description; - std::vector>> Keys; + std::vector>> Keys; std::function Unknown; Parser *Outer; @@ -79,7 +78,7 @@ // Parse is called when Key is encountered, and passed the associated value. // It should emit diagnostics if the value is invalid (e.g. wrong type). // If Key is seen twice, Parse runs only once and an error is reported. - void handle(llvm::StringLiteral Key, std::function Parse) { + void handle(llvm::StringLiteral Key, std::function Parse) { for (const auto &Entry : Keys) { (void) Entry; assert(Entry.first != Key && "duplicate key handler"); @@ -94,16 +93,17 @@ } // Process a mapping node and call handlers for each key/value pair. - bool parse(Node &N) const { + void parse(Node &N) const { if (N.getType() != Node::NK_Mapping) { Outer->error(Description + " should be a dictionary", N); - return false; + return; } llvm::SmallSet Seen; + // We *must* consume all items, even on error, or the parser will assert. for (auto &KV : llvm::cast(N)) { auto *K = KV.getKey(); if (!K) // YAMLParser emitted an error. - return false; + continue; auto Key = Outer->scalarValue(*K, "Dictionary key"); if (!Key) continue; @@ -113,13 +113,12 @@ } auto *Value = KV.getValue(); if (!Value) // YAMLParser emitted an error. - return false; + continue; bool Matched = false; for (const auto &Handler : Keys) { if (Handler.first == **Key) { - if (!Handler.second(*Value)) - return false; Matched = true; + Handler.second(*Value); break; } } @@ -129,7 +128,6 @@ Unknown(**Key); } } - return true; } }; @@ -154,6 +152,7 @@ } else if (auto *S = llvm::dyn_cast(&N)) { Result.emplace_back(S->getValue().str(), N.getSourceRange()); } else if (auto *S = llvm::dyn_cast(&N)) { + // We *must* consume all items, even on error, or the parser will assert. for (auto &Child : *S) { if (auto Value = scalarValue(Child, "List item")) Result.push_back(std::move(*Value)); @@ -167,6 +166,7 @@ // Report a "hard" error, reflecting a config file that can never be valid. void error(const llvm::Twine &Msg, const Node &N) { + HadError = true; SM.PrintMessage(N.getSourceRange().Start, llvm::SourceMgr::DK_Error, Msg, N.getSourceRange()); } @@ -196,7 +196,7 @@ &Diags); std::vector Result; for (auto &Doc : llvm::yaml::Stream(*Buf, *SM)) { - if (Node *N = Doc.parseBlockNode()) { + if (Node *N = Doc.getRoot()) { Fragment Fragment; Fragment.Source.Manager = SM; Fragment.Source.Location = N->getSourceRange().Start; diff --git a/clang-tools-extra/clangd/unittests/ConfigYAMLTests.cpp b/clang-tools-extra/clangd/unittests/ConfigYAMLTests.cpp --- a/clang-tools-extra/clangd/unittests/ConfigYAMLTests.cpp +++ b/clang-tools-extra/clangd/unittests/ConfigYAMLTests.cpp @@ -98,10 +98,25 @@ DiagKind(llvm::SourceMgr::DK_Error), DiagPos(YAML.point()), DiagRange(llvm::None)))); - ASSERT_EQ(Results.size(), 2u); + ASSERT_EQ(Results.size(), 1u); // invalid fragment discarded. EXPECT_THAT(Results.front().CompileFlags.Add, ElementsAre(Val("first"))); EXPECT_TRUE(Results.front().If.HasUnrecognizedCondition); - EXPECT_THAT(Results.back().CompileFlags.Add, IsEmpty()); +} + +TEST(ParseYAML, Invalid) { + CapturedDiags Diags; + const char *YAML = R"yaml( +If: + +horrible +--- +- 1 + )yaml"; + auto Results = Fragment::parseYAML(YAML, "config.yaml", Diags.callback()); + EXPECT_THAT(Diags.Diagnostics, + ElementsAre(DiagMessage("If should be a dictionary"), + DiagMessage("Config should be a dictionary"))); + ASSERT_THAT(Results, IsEmpty()); } } // namespace