[clangd] Drop diags from non-written #include.
Summary: This would fix that we show weird diagnostics on random lines of the main file.
Reviewers: ilya-biryukov
Subscribers: MaskRay, jkorous, arphaman, kadircet, cfe-commits
Tags: #clang
Differential Revision: https://reviews.llvm.org/D66074
git-svn-id: https://llvm.org/svn/llvm-project/clang-tools-extra/trunk@368549 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/clangd/Diagnostics.cpp b/clangd/Diagnostics.cpp
index 919182d..7f1ab06 100644
--- a/clangd/Diagnostics.cpp
+++ b/clangd/Diagnostics.cpp
@@ -122,8 +122,12 @@
return SM.getIncludeLoc(SM.getFileID(SLoc));
};
for (auto IncludeLocation = GetIncludeLoc(DiagLoc); IncludeLocation.isValid();
- IncludeLocation = GetIncludeLoc(IncludeLocation))
- IncludeInMainFile = IncludeLocation;
+ IncludeLocation = GetIncludeLoc(IncludeLocation)) {
+ if (clangd::isInsideMainFile(IncludeLocation, SM)) {
+ IncludeInMainFile = IncludeLocation;
+ break;
+ }
+ }
if (IncludeInMainFile.isInvalid())
return false;
diff --git a/clangd/unittests/DiagnosticsTests.cpp b/clangd/unittests/DiagnosticsTests.cpp
index c80dafe..2089c89 100644
--- a/clangd/unittests/DiagnosticsTests.cpp
+++ b/clangd/unittests/DiagnosticsTests.cpp
@@ -948,6 +948,15 @@
EXPECT_THAT(TU.build().getDiagnostics(), UnorderedElementsAre());
}
+TEST(IgnoreDiags, FromNonWrittenInclude) {
+ TestTU TU = TestTU::withCode("");
+ TU.ExtraArgs.push_back("--include=a.h");
+ TU.AdditionalFiles = {{"a.h", "void main();"}};
+ // The diagnostic "main must return int" is from the header, we don't attempt
+ // to render it in the main file as there is no written location there.
+ EXPECT_THAT(TU.build().getDiagnostics(), UnorderedElementsAre());
+}
+
} // namespace
} // namespace clangd