Author: Raphael Isemann Date: 2020-04-27T13:16:08+02:00 New Revision: 9f1e81f1c0ac40f81e2f398148f09d1852226240
URL: https://github.com/llvm/llvm-project/commit/9f1e81f1c0ac40f81e2f398148f09d1852226240 DIFF: https://github.com/llvm/llvm-project/commit/9f1e81f1c0ac40f81e2f398148f09d1852226240.diff LOG: [ASTImporter] Also import overwritten file buffers Summary: Overwritten file buffers are at the moment ignored when importing and instead only the underlying file buffer is imported. This patch fixes this by not going to the underlying file entry if the file has an overwritten buffer. Reviewers: martong, a.sidorin, shafik Reviewed By: martong, shafik Subscribers: rnkovacs Differential Revision: https://reviews.llvm.org/D78086 Added: Modified: clang/lib/AST/ASTImporter.cpp clang/unittests/AST/ASTImporterTest.cpp Removed: ################################################################################ diff --git a/clang/lib/AST/ASTImporter.cpp b/clang/lib/AST/ASTImporter.cpp index ecb5ce85c204..477c035d8a59 100644 --- a/clang/lib/AST/ASTImporter.cpp +++ b/clang/lib/AST/ASTImporter.cpp @@ -8560,7 +8560,7 @@ Expected<FileID> ASTImporter::Import(FileID FromID, bool IsBuiltin) { } else { const SrcMgr::ContentCache *Cache = FromSLoc.getFile().getContentCache(); - if (!IsBuiltin) { + if (!IsBuiltin && !Cache->BufferOverridden) { // Include location of this file. ExpectedSLoc ToIncludeLoc = Import(FromSLoc.getFile().getIncludeLoc()); if (!ToIncludeLoc) diff --git a/clang/unittests/AST/ASTImporterTest.cpp b/clang/unittests/AST/ASTImporterTest.cpp index 8ed6c487cb53..a390b49d1641 100644 --- a/clang/unittests/AST/ASTImporterTest.cpp +++ b/clang/unittests/AST/ASTImporterTest.cpp @@ -12,6 +12,7 @@ #include "clang/ASTMatchers/ASTMatchers.h" #include "llvm/ADT/StringMap.h" +#include "llvm/Support/SmallVectorMemoryBuffer.h" #include "clang/AST/DeclContextInternals.h" #include "gtest/gtest.h" @@ -5896,6 +5897,61 @@ TEST_P(ImportSourceLocations, PreserveFileIDTreeStructure) { EXPECT_FALSE(ToSM.isBeforeInTranslationUnit(Location2, Location1)); } +TEST_P(ImportSourceLocations, NormalFileBuffer) { + // Test importing normal file buffers. + + std::string Path = "input0.c"; + std::string Source = "int X;"; + TranslationUnitDecl *FromTU = getTuDecl(Source, Lang_C, Path); + + SourceLocation ImportedLoc; + { + // Import the VarDecl to trigger the importing of the FileID. + auto Pattern = varDecl(hasName("X")); + VarDecl *FromD = FirstDeclMatcher<VarDecl>().match(FromTU, Pattern); + ImportedLoc = Import(FromD, Lang_C)->getLocation(); + } + + // Make sure the imported buffer has the original contents. + SourceManager &ToSM = ToAST->getSourceManager(); + FileID ImportedID = ToSM.getFileID(ImportedLoc); + EXPECT_EQ(Source, ToSM.getBuffer(ImportedID, SourceLocation())->getBuffer()); +} + +TEST_P(ImportSourceLocations, OverwrittenFileBuffer) { + // Test importing overwritten file buffers. + + std::string Path = "input0.c"; + TranslationUnitDecl *FromTU = getTuDecl("int X;", Lang_C, Path); + + // Overwrite the file buffer for our input file with new content. + const std::string Contents = "overwritten contents"; + SourceLocation ImportedLoc; + { + SourceManager &FromSM = FromTU->getASTContext().getSourceManager(); + clang::FileManager &FM = FromSM.getFileManager(); + const clang::FileEntry &FE = + *FM.getVirtualFile(Path, static_cast<off_t>(Contents.size()), 0); + + llvm::SmallVector<char, 64> Buffer; + Buffer.append(Contents.begin(), Contents.end()); + auto FileContents = + std::make_unique<llvm::SmallVectorMemoryBuffer>(std::move(Buffer), Path); + FromSM.overrideFileContents(&FE, std::move(FileContents)); + + // Import the VarDecl to trigger the importing of the FileID. + auto Pattern = varDecl(hasName("X")); + VarDecl *FromD = FirstDeclMatcher<VarDecl>().match(FromTU, Pattern); + ImportedLoc = Import(FromD, Lang_C)->getLocation(); + } + + // Make sure the imported buffer has the overwritten contents. + SourceManager &ToSM = ToAST->getSourceManager(); + FileID ImportedID = ToSM.getFileID(ImportedLoc); + EXPECT_EQ(Contents, + ToSM.getBuffer(ImportedID, SourceLocation())->getBuffer()); +} + TEST_P(ASTImporterOptionSpecificTestBase, ImportExprOfAlignmentAttr) { // Test if import of these packed and aligned attributes does not trigger an // error situation where source location from 'From' context is referenced in _______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits