diff --git a/clang/lib/Tooling/Tooling.cpp b/clang/lib/Tooling/Tooling.cpp --- a/clang/lib/Tooling/Tooling.cpp +++ b/clang/lib/Tooling/Tooling.cpp @@ -354,16 +354,16 @@ } TextDiagnosticPrinter DiagnosticPrinter(llvm::errs(), DiagOpts); - DiagnosticsEngine Diagnostics( - IntrusiveRefCntPtr(new DiagnosticIDs()), &*DiagOpts, - DiagConsumer ? DiagConsumer : &DiagnosticPrinter, false); + IntrusiveRefCntPtr Diagnostics = + CompilerInstance::createDiagnostics( + &*DiagOpts, DiagConsumer ? DiagConsumer : &DiagnosticPrinter, false); // Although `Diagnostics` are used only for command-line parsing, the custom // `DiagConsumer` might expect a `SourceManager` to be present. - SourceManager SrcMgr(Diagnostics, *Files); - Diagnostics.setSourceManager(&SrcMgr); + SourceManager SrcMgr(*Diagnostics, *Files); + Diagnostics->setSourceManager(&SrcMgr); const std::unique_ptr Driver( - newDriver(&Diagnostics, BinaryName, &Files->getVirtualFileSystem())); + newDriver(&*Diagnostics, BinaryName, &Files->getVirtualFileSystem())); // The "input file not found" diagnostics from the driver are useful. // The driver is only aware of the VFS working directory, but some clients // change this at the FileManager level instead. @@ -375,11 +375,11 @@ if (!Compilation) return false; const llvm::opt::ArgStringList *const CC1Args = getCC1Arguments( - &Diagnostics, Compilation.get()); + &*Diagnostics, Compilation.get()); if (!CC1Args) return false; std::unique_ptr Invocation( - newInvocation(&Diagnostics, *CC1Args, BinaryName)); + newInvocation(&*Diagnostics, *CC1Args, BinaryName)); return runInvocation(BinaryName, Compilation.get(), std::move(Invocation), std::move(PCHContainerOps)); } diff --git a/clang/unittests/Tooling/ToolingTest.cpp b/clang/unittests/Tooling/ToolingTest.cpp --- a/clang/unittests/Tooling/ToolingTest.cpp +++ b/clang/unittests/Tooling/ToolingTest.cpp @@ -224,6 +224,82 @@ EXPECT_TRUE(Invocation.run()); } +struct ErrorCountingDiagnosticConsumer : public DiagnosticConsumer { + ErrorCountingDiagnosticConsumer() : NumErrorsSeen(0) {} + void HandleDiagnostic(DiagnosticsEngine::Level DiagLevel, + const Diagnostic &Info) override { + if (DiagLevel == DiagnosticsEngine::Level::Error) + ++NumErrorsSeen; + } + unsigned NumErrorsSeen; +}; + +TEST(ToolInvocation, DiagnosticsEngineProperlyInitializedForCC1Construction) { + llvm::IntrusiveRefCntPtr OverlayFileSystem( + new llvm::vfs::OverlayFileSystem(llvm::vfs::getRealFileSystem())); + llvm::IntrusiveRefCntPtr InMemoryFileSystem( + new llvm::vfs::InMemoryFileSystem); + OverlayFileSystem->pushOverlay(InMemoryFileSystem); + llvm::IntrusiveRefCntPtr Files( + new FileManager(FileSystemOptions(), OverlayFileSystem)); + + std::vector Args; + Args.push_back("tool-executable"); + Args.push_back("-target"); + // Invalid argument that by default results in an error diagnostic: + Args.push_back("i386-apple-ios14.0-simulator"); + // Argument that downgrades the error into a warning: + Args.push_back("-Wno-error=invalid-ios-deployment-target"); + Args.push_back("-fsyntax-only"); + Args.push_back("test.cpp"); + + clang::tooling::ToolInvocation Invocation( + Args, std::make_unique(), Files.get()); + InMemoryFileSystem->addFile( + "test.cpp", 0, llvm::MemoryBuffer::getMemBuffer("int a() {}\n")); + ErrorCountingDiagnosticConsumer Consumer; + Invocation.setDiagnosticConsumer(&Consumer); + EXPECT_TRUE(Invocation.run()); + // Check that `-Wno-error=invalid-ios-deployment-target` downgraded the error + // into a warning. + EXPECT_EQ(Consumer.NumErrorsSeen, 0u); +} + +TEST(ToolInvocation, CustomDiagnosticOptionsOverwriteParsedOnes) { + llvm::IntrusiveRefCntPtr OverlayFileSystem( + new llvm::vfs::OverlayFileSystem(llvm::vfs::getRealFileSystem())); + llvm::IntrusiveRefCntPtr InMemoryFileSystem( + new llvm::vfs::InMemoryFileSystem); + OverlayFileSystem->pushOverlay(InMemoryFileSystem); + llvm::IntrusiveRefCntPtr Files( + new FileManager(FileSystemOptions(), OverlayFileSystem)); + + std::vector Args; + Args.push_back("tool-executable"); + Args.push_back("-target"); + // Invalid argument that by default results in an error diagnostic: + Args.push_back("i386-apple-ios14.0-simulator"); + // Argument that downgrades the error into a warning: + Args.push_back("-Wno-error=invalid-ios-deployment-target"); + Args.push_back("-fsyntax-only"); + Args.push_back("test.cpp"); + + clang::tooling::ToolInvocation Invocation( + Args, std::make_unique(), Files.get()); + InMemoryFileSystem->addFile( + "test.cpp", 0, llvm::MemoryBuffer::getMemBuffer("int a() {}\n")); + ErrorCountingDiagnosticConsumer Consumer; + Invocation.setDiagnosticConsumer(&Consumer); + + auto DiagOpts = llvm::makeIntrusiveRefCnt(); + Invocation.setDiagnosticOptions(&*DiagOpts); + + EXPECT_TRUE(Invocation.run()); + // Check that `-Wno-error=invalid-ios-deployment-target` didn't downgrade the + // error into a warning due to being overwritten by custom diagnostic options. + EXPECT_EQ(Consumer.NumErrorsSeen, 1u); +} + struct DiagnosticConsumerExpectingSourceManager : public DiagnosticConsumer { bool SawSourceManager;