diff options
Diffstat (limited to 'gnu/llvm/tools/clang/lib/CrossTU/CrossTranslationUnit.cpp')
| -rw-r--r-- | gnu/llvm/tools/clang/lib/CrossTU/CrossTranslationUnit.cpp | 365 |
1 files changed, 0 insertions, 365 deletions
diff --git a/gnu/llvm/tools/clang/lib/CrossTU/CrossTranslationUnit.cpp b/gnu/llvm/tools/clang/lib/CrossTU/CrossTranslationUnit.cpp deleted file mode 100644 index 7c97beb498a..00000000000 --- a/gnu/llvm/tools/clang/lib/CrossTU/CrossTranslationUnit.cpp +++ /dev/null @@ -1,365 +0,0 @@ -//===--- CrossTranslationUnit.cpp - -----------------------------*- C++ -*-===// -// -// The LLVM Compiler Infrastructure -// -// This file is distributed under the University of Illinois Open Source -// License. See LICENSE.TXT for details. -// -//===----------------------------------------------------------------------===// -// -// This file implements the CrossTranslationUnit interface. -// -//===----------------------------------------------------------------------===// -#include "clang/CrossTU/CrossTranslationUnit.h" -#include "clang/AST/ASTImporter.h" -#include "clang/AST/Decl.h" -#include "clang/Basic/TargetInfo.h" -#include "clang/CrossTU/CrossTUDiagnostic.h" -#include "clang/Frontend/ASTUnit.h" -#include "clang/Frontend/CompilerInstance.h" -#include "clang/Frontend/TextDiagnosticPrinter.h" -#include "clang/Index/USRGeneration.h" -#include "llvm/ADT/Triple.h" -#include "llvm/ADT/Statistic.h" -#include "llvm/Support/ErrorHandling.h" -#include "llvm/Support/ManagedStatic.h" -#include "llvm/Support/Path.h" -#include "llvm/Support/raw_ostream.h" -#include <fstream> -#include <sstream> - -namespace clang { -namespace cross_tu { - -namespace { - -#define DEBUG_TYPE "CrossTranslationUnit" -STATISTIC(NumGetCTUCalled, "The # of getCTUDefinition function called"); -STATISTIC( - NumNotInOtherTU, - "The # of getCTUDefinition called but the function is not in any other TU"); -STATISTIC(NumGetCTUSuccess, - "The # of getCTUDefinition successfully returned the " - "requested function's body"); -STATISTIC(NumTripleMismatch, "The # of triple mismatches"); -STATISTIC(NumLangMismatch, "The # of language mismatches"); - -// Same as Triple's equality operator, but we check a field only if that is -// known in both instances. -bool hasEqualKnownFields(const llvm::Triple &Lhs, const llvm::Triple &Rhs) { - using llvm::Triple; - if (Lhs.getArch() != Triple::UnknownArch && - Rhs.getArch() != Triple::UnknownArch && Lhs.getArch() != Rhs.getArch()) - return false; - if (Lhs.getSubArch() != Triple::NoSubArch && - Rhs.getSubArch() != Triple::NoSubArch && - Lhs.getSubArch() != Rhs.getSubArch()) - return false; - if (Lhs.getVendor() != Triple::UnknownVendor && - Rhs.getVendor() != Triple::UnknownVendor && - Lhs.getVendor() != Rhs.getVendor()) - return false; - if (!Lhs.isOSUnknown() && !Rhs.isOSUnknown() && - Lhs.getOS() != Rhs.getOS()) - return false; - if (Lhs.getEnvironment() != Triple::UnknownEnvironment && - Rhs.getEnvironment() != Triple::UnknownEnvironment && - Lhs.getEnvironment() != Rhs.getEnvironment()) - return false; - if (Lhs.getObjectFormat() != Triple::UnknownObjectFormat && - Rhs.getObjectFormat() != Triple::UnknownObjectFormat && - Lhs.getObjectFormat() != Rhs.getObjectFormat()) - return false; - return true; -} - -// FIXME: This class is will be removed after the transition to llvm::Error. -class IndexErrorCategory : public std::error_category { -public: - const char *name() const noexcept override { return "clang.index"; } - - std::string message(int Condition) const override { - switch (static_cast<index_error_code>(Condition)) { - case index_error_code::unspecified: - return "An unknown error has occurred."; - case index_error_code::missing_index_file: - return "The index file is missing."; - case index_error_code::invalid_index_format: - return "Invalid index file format."; - case index_error_code::multiple_definitions: - return "Multiple definitions in the index file."; - case index_error_code::missing_definition: - return "Missing definition from the index file."; - case index_error_code::failed_import: - return "Failed to import the definition."; - case index_error_code::failed_to_get_external_ast: - return "Failed to load external AST source."; - case index_error_code::failed_to_generate_usr: - return "Failed to generate USR."; - case index_error_code::triple_mismatch: - return "Triple mismatch"; - case index_error_code::lang_mismatch: - return "Language mismatch"; - } - llvm_unreachable("Unrecognized index_error_code."); - } -}; - -static llvm::ManagedStatic<IndexErrorCategory> Category; -} // end anonymous namespace - -char IndexError::ID; - -void IndexError::log(raw_ostream &OS) const { - OS << Category->message(static_cast<int>(Code)) << '\n'; -} - -std::error_code IndexError::convertToErrorCode() const { - return std::error_code(static_cast<int>(Code), *Category); -} - -llvm::Expected<llvm::StringMap<std::string>> -parseCrossTUIndex(StringRef IndexPath, StringRef CrossTUDir) { - std::ifstream ExternalMapFile(IndexPath); - if (!ExternalMapFile) - return llvm::make_error<IndexError>(index_error_code::missing_index_file, - IndexPath.str()); - - llvm::StringMap<std::string> Result; - std::string Line; - unsigned LineNo = 1; - while (std::getline(ExternalMapFile, Line)) { - const size_t Pos = Line.find(" "); - if (Pos > 0 && Pos != std::string::npos) { - StringRef LineRef{Line}; - StringRef LookupName = LineRef.substr(0, Pos); - if (Result.count(LookupName)) - return llvm::make_error<IndexError>( - index_error_code::multiple_definitions, IndexPath.str(), LineNo); - StringRef FileName = LineRef.substr(Pos + 1); - SmallString<256> FilePath = CrossTUDir; - llvm::sys::path::append(FilePath, FileName); - Result[LookupName] = FilePath.str().str(); - } else - return llvm::make_error<IndexError>( - index_error_code::invalid_index_format, IndexPath.str(), LineNo); - LineNo++; - } - return Result; -} - -std::string -createCrossTUIndexString(const llvm::StringMap<std::string> &Index) { - std::ostringstream Result; - for (const auto &E : Index) - Result << E.getKey().str() << " " << E.getValue() << '\n'; - return Result.str(); -} - -CrossTranslationUnitContext::CrossTranslationUnitContext(CompilerInstance &CI) - : CI(CI), Context(CI.getASTContext()) {} - -CrossTranslationUnitContext::~CrossTranslationUnitContext() {} - -std::string CrossTranslationUnitContext::getLookupName(const NamedDecl *ND) { - SmallString<128> DeclUSR; - bool Ret = index::generateUSRForDecl(ND, DeclUSR); (void)Ret; - assert(!Ret && "Unable to generate USR"); - return DeclUSR.str(); -} - -/// Recursively visits the function decls of a DeclContext, and looks up a -/// function based on USRs. -const FunctionDecl * -CrossTranslationUnitContext::findFunctionInDeclContext(const DeclContext *DC, - StringRef LookupFnName) { - assert(DC && "Declaration Context must not be null"); - for (const Decl *D : DC->decls()) { - const auto *SubDC = dyn_cast<DeclContext>(D); - if (SubDC) - if (const auto *FD = findFunctionInDeclContext(SubDC, LookupFnName)) - return FD; - - const auto *ND = dyn_cast<FunctionDecl>(D); - const FunctionDecl *ResultDecl; - if (!ND || !ND->hasBody(ResultDecl)) - continue; - if (getLookupName(ResultDecl) != LookupFnName) - continue; - return ResultDecl; - } - return nullptr; -} - -llvm::Expected<const FunctionDecl *> -CrossTranslationUnitContext::getCrossTUDefinition(const FunctionDecl *FD, - StringRef CrossTUDir, - StringRef IndexName, - bool DisplayCTUProgress) { - assert(FD && "FD is missing, bad call to this function!"); - assert(!FD->hasBody() && "FD has a definition in current translation unit!"); - ++NumGetCTUCalled; - const std::string LookupFnName = getLookupName(FD); - if (LookupFnName.empty()) - return llvm::make_error<IndexError>( - index_error_code::failed_to_generate_usr); - llvm::Expected<ASTUnit *> ASTUnitOrError = - loadExternalAST(LookupFnName, CrossTUDir, IndexName, DisplayCTUProgress); - if (!ASTUnitOrError) - return ASTUnitOrError.takeError(); - ASTUnit *Unit = *ASTUnitOrError; - assert(&Unit->getFileManager() == - &Unit->getASTContext().getSourceManager().getFileManager()); - - const llvm::Triple &TripleTo = Context.getTargetInfo().getTriple(); - const llvm::Triple &TripleFrom = - Unit->getASTContext().getTargetInfo().getTriple(); - // The imported AST had been generated for a different target. - // Some parts of the triple in the loaded ASTContext can be unknown while the - // very same parts in the target ASTContext are known. Thus we check for the - // known parts only. - if (!hasEqualKnownFields(TripleTo, TripleFrom)) { - // TODO: Pass the SourceLocation of the CallExpression for more precise - // diagnostics. - ++NumTripleMismatch; - return llvm::make_error<IndexError>(index_error_code::triple_mismatch, - Unit->getMainFileName(), TripleTo.str(), - TripleFrom.str()); - } - - const auto &LangTo = Context.getLangOpts(); - const auto &LangFrom = Unit->getASTContext().getLangOpts(); - // FIXME: Currenty we do not support CTU across C++ and C and across - // different dialects of C++. - if (LangTo.CPlusPlus != LangFrom.CPlusPlus) { - ++NumLangMismatch; - return llvm::make_error<IndexError>(index_error_code::lang_mismatch); - } - - TranslationUnitDecl *TU = Unit->getASTContext().getTranslationUnitDecl(); - if (const FunctionDecl *ResultDecl = - findFunctionInDeclContext(TU, LookupFnName)) - return importDefinition(ResultDecl); - return llvm::make_error<IndexError>(index_error_code::failed_import); -} - -void CrossTranslationUnitContext::emitCrossTUDiagnostics(const IndexError &IE) { - switch (IE.getCode()) { - case index_error_code::missing_index_file: - Context.getDiagnostics().Report(diag::err_ctu_error_opening) - << IE.getFileName(); - break; - case index_error_code::invalid_index_format: - Context.getDiagnostics().Report(diag::err_extdefmap_parsing) - << IE.getFileName() << IE.getLineNum(); - break; - case index_error_code::multiple_definitions: - Context.getDiagnostics().Report(diag::err_multiple_def_index) - << IE.getLineNum(); - break; - case index_error_code::triple_mismatch: - Context.getDiagnostics().Report(diag::warn_ctu_incompat_triple) - << IE.getFileName() << IE.getTripleToName() << IE.getTripleFromName(); - break; - default: - break; - } -} - -llvm::Expected<ASTUnit *> CrossTranslationUnitContext::loadExternalAST( - StringRef LookupName, StringRef CrossTUDir, StringRef IndexName, - bool DisplayCTUProgress) { - // FIXME: The current implementation only supports loading functions with - // a lookup name from a single translation unit. If multiple - // translation units contains functions with the same lookup name an - // error will be returned. - ASTUnit *Unit = nullptr; - auto FnUnitCacheEntry = FunctionASTUnitMap.find(LookupName); - if (FnUnitCacheEntry == FunctionASTUnitMap.end()) { - if (FunctionFileMap.empty()) { - SmallString<256> IndexFile = CrossTUDir; - if (llvm::sys::path::is_absolute(IndexName)) - IndexFile = IndexName; - else - llvm::sys::path::append(IndexFile, IndexName); - llvm::Expected<llvm::StringMap<std::string>> IndexOrErr = - parseCrossTUIndex(IndexFile, CrossTUDir); - if (IndexOrErr) - FunctionFileMap = *IndexOrErr; - else - return IndexOrErr.takeError(); - } - - auto It = FunctionFileMap.find(LookupName); - if (It == FunctionFileMap.end()) { - ++NumNotInOtherTU; - return llvm::make_error<IndexError>(index_error_code::missing_definition); - } - StringRef ASTFileName = It->second; - auto ASTCacheEntry = FileASTUnitMap.find(ASTFileName); - if (ASTCacheEntry == FileASTUnitMap.end()) { - IntrusiveRefCntPtr<DiagnosticOptions> DiagOpts = new DiagnosticOptions(); - TextDiagnosticPrinter *DiagClient = - new TextDiagnosticPrinter(llvm::errs(), &*DiagOpts); - IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); - IntrusiveRefCntPtr<DiagnosticsEngine> Diags( - new DiagnosticsEngine(DiagID, &*DiagOpts, DiagClient)); - - std::unique_ptr<ASTUnit> LoadedUnit(ASTUnit::LoadFromASTFile( - ASTFileName, CI.getPCHContainerOperations()->getRawReader(), - ASTUnit::LoadEverything, Diags, CI.getFileSystemOpts())); - Unit = LoadedUnit.get(); - FileASTUnitMap[ASTFileName] = std::move(LoadedUnit); - if (DisplayCTUProgress) { - llvm::errs() << "CTU loaded AST file: " - << ASTFileName << "\n"; - } - } else { - Unit = ASTCacheEntry->second.get(); - } - FunctionASTUnitMap[LookupName] = Unit; - } else { - Unit = FnUnitCacheEntry->second; - } - if (!Unit) - return llvm::make_error<IndexError>( - index_error_code::failed_to_get_external_ast); - return Unit; -} - -llvm::Expected<const FunctionDecl *> -CrossTranslationUnitContext::importDefinition(const FunctionDecl *FD) { - assert(FD->hasBody() && "Functions to be imported should have body."); - - ASTImporter &Importer = getOrCreateASTImporter(FD->getASTContext()); - auto *ToDecl = - cast_or_null<FunctionDecl>(Importer.Import(const_cast<FunctionDecl *>(FD))); - if (!ToDecl) - return llvm::make_error<IndexError>(index_error_code::failed_import); - assert(ToDecl->hasBody()); - assert(FD->hasBody() && "Functions already imported should have body."); - ++NumGetCTUSuccess; - return ToDecl; -} - -void CrossTranslationUnitContext::lazyInitLookupTable( - TranslationUnitDecl *ToTU) { - if (!LookupTable) - LookupTable = llvm::make_unique<ASTImporterLookupTable>(*ToTU); -} - -ASTImporter & -CrossTranslationUnitContext::getOrCreateASTImporter(ASTContext &From) { - auto I = ASTUnitImporterMap.find(From.getTranslationUnitDecl()); - if (I != ASTUnitImporterMap.end()) - return *I->second; - lazyInitLookupTable(Context.getTranslationUnitDecl()); - ASTImporter *NewImporter = new ASTImporter( - Context, Context.getSourceManager().getFileManager(), From, - From.getSourceManager().getFileManager(), false, LookupTable.get()); - ASTUnitImporterMap[From.getTranslationUnitDecl()].reset(NewImporter); - return *NewImporter; -} - -} // namespace cross_tu -} // namespace clang |