diff --git a/clang/lib/Frontend/PrecompiledPreamble.cpp b/clang/lib/Frontend/PrecompiledPreamble.cpp --- a/clang/lib/Frontend/PrecompiledPreamble.cpp +++ b/clang/lib/Frontend/PrecompiledPreamble.cpp @@ -234,9 +234,10 @@ class PrecompilePreambleAction : public ASTFrontendAction { public: - PrecompilePreambleAction(std::string *InMemStorage, + PrecompilePreambleAction(std::shared_ptr Buffer, bool WritePCHFile, PreambleCallbacks &Callbacks) - : InMemStorage(InMemStorage), Callbacks(Callbacks) {} + : Buffer(std::move(Buffer)), WritePCHFile(WritePCHFile), + Callbacks(Callbacks) {} std::unique_ptr CreateASTConsumer(CompilerInstance &CI, StringRef InFile) override; @@ -244,6 +245,12 @@ bool hasEmittedPreamblePCH() const { return HasEmittedPreamblePCH; } void setEmittedPreamblePCH(ASTWriter &Writer) { + if (FileOS) { + *FileOS << Buffer->Data; + // Make sure it hits disk now. + FileOS->flush(); + } + this->HasEmittedPreamblePCH = true; Callbacks.AfterPCHEmitted(Writer); } @@ -262,7 +269,9 @@ friend class PrecompilePreambleConsumer; bool HasEmittedPreamblePCH = false; - std::string *InMemStorage; + std::shared_ptr Buffer; + bool WritePCHFile; // otherwise the PCH is written into the PCHBuffer only. + std::unique_ptr FileOS; // null if in-memory PreambleCallbacks &Callbacks; }; @@ -272,12 +281,11 @@ const Preprocessor &PP, InMemoryModuleCache &ModuleCache, StringRef isysroot, - std::unique_ptr Out) - : PCHGenerator(PP, ModuleCache, "", isysroot, - std::make_shared(), + std::shared_ptr Buffer) + : PCHGenerator(PP, ModuleCache, "", isysroot, std::move(Buffer), ArrayRef>(), /*AllowASTWithErrors=*/true), - Action(Action), Out(std::move(Out)) {} + Action(Action) {} bool HandleTopLevelDecl(DeclGroupRef DG) override { Action.Callbacks.HandleTopLevelDecl(DG); @@ -288,15 +296,6 @@ PCHGenerator::HandleTranslationUnit(Ctx); if (!hasEmittedPCH()) return; - - // Write the generated bitstream to "Out". - *Out << getPCH(); - // Make sure it hits disk now. - Out->flush(); - // Free the buffer. - llvm::SmallVector Empty; - getPCH() = std::move(Empty); - Action.setEmittedPreamblePCH(getWriter()); } @@ -306,7 +305,6 @@ private: PrecompilePreambleAction &Action; - std::unique_ptr Out; }; std::unique_ptr @@ -316,21 +314,18 @@ if (!GeneratePCHAction::ComputeASTConsumerArguments(CI, Sysroot)) return nullptr; - std::unique_ptr OS; - if (InMemStorage) { - OS = std::make_unique(*InMemStorage); - } else { - std::string OutputFile; - OS = GeneratePCHAction::CreateOutputFile(CI, InFile, OutputFile); + if (WritePCHFile) { + std::string OutputFile; // unused + FileOS = GeneratePCHAction::CreateOutputFile(CI, InFile, OutputFile); + if (!FileOS) + return nullptr; } - if (!OS) - return nullptr; if (!CI.getFrontendOpts().RelocatablePCH) Sysroot.clear(); return std::make_unique( - *this, CI.getPreprocessor(), CI.getModuleCache(), Sysroot, std::move(OS)); + *this, CI.getPreprocessor(), CI.getModuleCache(), Sysroot, Buffer); } template bool moveOnNoError(llvm::ErrorOr Val, T &Output) { @@ -356,9 +351,9 @@ S->File = std::move(File); return S; } - static std::unique_ptr inMemory() { + static std::unique_ptr inMemory(std::shared_ptr Buf) { std::unique_ptr S(new PCHStorage()); - S->Memory.emplace(); + S->Memory = std::move(Buf); return S; } @@ -371,15 +366,16 @@ llvm_unreachable("Neither Memory nor File?"); } llvm::StringRef filePath() const { return File->getFilePath(); } - llvm::StringRef memoryContents() const { return *Memory; } - std::string &memoryBufferForWrite() { return *Memory; } + llvm::StringRef memoryContents() const { + return StringRef(Memory->Data.data(), Memory->Data.size()); + } private: PCHStorage() = default; PCHStorage(const PCHStorage &) = delete; PCHStorage &operator=(const PCHStorage &) = delete; - llvm::Optional Memory; + std::shared_ptr Memory; std::unique_ptr File; }; @@ -402,9 +398,10 @@ PreprocessorOptions &PreprocessorOpts = PreambleInvocation->getPreprocessorOpts(); + std::shared_ptr Buffer = std::make_shared(); std::unique_ptr Storage; if (StoreInMemory) { - Storage = PCHStorage::inMemory(); + Storage = PCHStorage::inMemory(Buffer); } else { // Create a temporary file for the precompiled preamble. In rare // circumstances, this can fail. @@ -486,9 +483,10 @@ PreambleInputBuffer.release()); } - std::unique_ptr Act; - Act.reset(new PrecompilePreambleAction( - StoreInMemory ? &Storage->memoryBufferForWrite() : nullptr, Callbacks)); + auto Act = std::make_unique( + std::move(Buffer), + /*WritePCHFile=*/Storage->getKind() == PCHStorage::Kind::TempFile, + Callbacks); if (!Act->BeginSourceFile(*Clang.get(), Clang->getFrontendOpts().Inputs[0])) return BuildPreambleError::BeginSourceFileFailed; @@ -518,6 +516,7 @@ if (!Act->hasEmittedPreamblePCH()) return BuildPreambleError::CouldntEmitPCH; + Act.reset(); // Frees the PCH buffer frees, unless Storage keeps it in memory. // Keep track of all of the files that the source manager knows about, // so we can verify whether they have changed or not.