1*0b57cec5SDimitry Andric //===- InMemoryModuleCache.cpp - Cache for loaded memory buffers ----------===// 2*0b57cec5SDimitry Andric // 3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*0b57cec5SDimitry Andric // 7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===// 8*0b57cec5SDimitry Andric 9*0b57cec5SDimitry Andric #include "clang/Serialization/InMemoryModuleCache.h" 10*0b57cec5SDimitry Andric #include "llvm/Support/MemoryBuffer.h" 11*0b57cec5SDimitry Andric 12*0b57cec5SDimitry Andric using namespace clang; 13*0b57cec5SDimitry Andric 14*0b57cec5SDimitry Andric InMemoryModuleCache::State getPCMState(llvm::StringRef Filename) const15*0b57cec5SDimitry AndricInMemoryModuleCache::getPCMState(llvm::StringRef Filename) const { 16*0b57cec5SDimitry Andric auto I = PCMs.find(Filename); 17*0b57cec5SDimitry Andric if (I == PCMs.end()) 18*0b57cec5SDimitry Andric return Unknown; 19*0b57cec5SDimitry Andric if (I->second.IsFinal) 20*0b57cec5SDimitry Andric return Final; 21*0b57cec5SDimitry Andric return I->second.Buffer ? Tentative : ToBuild; 22*0b57cec5SDimitry Andric } 23*0b57cec5SDimitry Andric 24*0b57cec5SDimitry Andric llvm::MemoryBuffer & addPCM(llvm::StringRef Filename,std::unique_ptr<llvm::MemoryBuffer> Buffer)25*0b57cec5SDimitry AndricInMemoryModuleCache::addPCM(llvm::StringRef Filename, 26*0b57cec5SDimitry Andric std::unique_ptr<llvm::MemoryBuffer> Buffer) { 27*0b57cec5SDimitry Andric auto Insertion = PCMs.insert(std::make_pair(Filename, std::move(Buffer))); 28*0b57cec5SDimitry Andric assert(Insertion.second && "Already has a PCM"); 29*0b57cec5SDimitry Andric return *Insertion.first->second.Buffer; 30*0b57cec5SDimitry Andric } 31*0b57cec5SDimitry Andric 32*0b57cec5SDimitry Andric llvm::MemoryBuffer & addBuiltPCM(llvm::StringRef Filename,std::unique_ptr<llvm::MemoryBuffer> Buffer)33*0b57cec5SDimitry AndricInMemoryModuleCache::addBuiltPCM(llvm::StringRef Filename, 34*0b57cec5SDimitry Andric std::unique_ptr<llvm::MemoryBuffer> Buffer) { 35*0b57cec5SDimitry Andric auto &PCM = PCMs[Filename]; 36*0b57cec5SDimitry Andric assert(!PCM.IsFinal && "Trying to override finalized PCM?"); 37*0b57cec5SDimitry Andric assert(!PCM.Buffer && "Trying to override tentative PCM?"); 38*0b57cec5SDimitry Andric PCM.Buffer = std::move(Buffer); 39*0b57cec5SDimitry Andric PCM.IsFinal = true; 40*0b57cec5SDimitry Andric return *PCM.Buffer; 41*0b57cec5SDimitry Andric } 42*0b57cec5SDimitry Andric 43*0b57cec5SDimitry Andric llvm::MemoryBuffer * lookupPCM(llvm::StringRef Filename) const44*0b57cec5SDimitry AndricInMemoryModuleCache::lookupPCM(llvm::StringRef Filename) const { 45*0b57cec5SDimitry Andric auto I = PCMs.find(Filename); 46*0b57cec5SDimitry Andric if (I == PCMs.end()) 47*0b57cec5SDimitry Andric return nullptr; 48*0b57cec5SDimitry Andric return I->second.Buffer.get(); 49*0b57cec5SDimitry Andric } 50*0b57cec5SDimitry Andric isPCMFinal(llvm::StringRef Filename) const51*0b57cec5SDimitry Andricbool InMemoryModuleCache::isPCMFinal(llvm::StringRef Filename) const { 52*0b57cec5SDimitry Andric return getPCMState(Filename) == Final; 53*0b57cec5SDimitry Andric } 54*0b57cec5SDimitry Andric shouldBuildPCM(llvm::StringRef Filename) const55*0b57cec5SDimitry Andricbool InMemoryModuleCache::shouldBuildPCM(llvm::StringRef Filename) const { 56*0b57cec5SDimitry Andric return getPCMState(Filename) == ToBuild; 57*0b57cec5SDimitry Andric } 58*0b57cec5SDimitry Andric tryToDropPCM(llvm::StringRef Filename)59*0b57cec5SDimitry Andricbool InMemoryModuleCache::tryToDropPCM(llvm::StringRef Filename) { 60*0b57cec5SDimitry Andric auto I = PCMs.find(Filename); 61*0b57cec5SDimitry Andric assert(I != PCMs.end() && "PCM to remove is unknown..."); 62*0b57cec5SDimitry Andric 63*0b57cec5SDimitry Andric auto &PCM = I->second; 64*0b57cec5SDimitry Andric assert(PCM.Buffer && "PCM to remove is scheduled to be built..."); 65*0b57cec5SDimitry Andric 66*0b57cec5SDimitry Andric if (PCM.IsFinal) 67*0b57cec5SDimitry Andric return true; 68*0b57cec5SDimitry Andric 69*0b57cec5SDimitry Andric PCM.Buffer.reset(); 70*0b57cec5SDimitry Andric return false; 71*0b57cec5SDimitry Andric } 72*0b57cec5SDimitry Andric finalizePCM(llvm::StringRef Filename)73*0b57cec5SDimitry Andricvoid InMemoryModuleCache::finalizePCM(llvm::StringRef Filename) { 74*0b57cec5SDimitry Andric auto I = PCMs.find(Filename); 75*0b57cec5SDimitry Andric assert(I != PCMs.end() && "PCM to finalize is unknown..."); 76*0b57cec5SDimitry Andric 77*0b57cec5SDimitry Andric auto &PCM = I->second; 78*0b57cec5SDimitry Andric assert(PCM.Buffer && "Trying to finalize a dropped PCM..."); 79*0b57cec5SDimitry Andric PCM.IsFinal = true; 80*0b57cec5SDimitry Andric } 81