1//===-- LLVMContext.cpp - Implement LLVMContext -----------------------===// 2// 3// The LLVM Compiler Infrastructure 4// 5// This file is distributed under the University of Illinois Open Source 6// License. See LICENSE.TXT for details. 7// 8//===----------------------------------------------------------------------===// 9// 10// This file implements LLVMContext, as a wrapper around the opaque 11// class LLVMContextImpl. 12// 13//===----------------------------------------------------------------------===// 14 15#include "llvm/LLVMContext.h" 16#include "llvm/Metadata.h" 17#include "llvm/Constants.h" 18#include "llvm/Instruction.h" 19#include "llvm/Support/ManagedStatic.h" 20#include "llvm/Support/SourceMgr.h" 21#include "LLVMContextImpl.h" 22#include <cctype> 23using namespace llvm; 24 25static ManagedStatic<LLVMContext> GlobalContext; 26 27LLVMContext& llvm::getGlobalContext() { 28 return *GlobalContext; 29} 30 31LLVMContext::LLVMContext() : pImpl(new LLVMContextImpl(*this)) { 32 // Create the fixed metadata kinds. This is done in the same order as the 33 // MD_* enum values so that they correspond. 34 35 // Create the 'dbg' metadata kind. 36 unsigned DbgID = getMDKindID("dbg"); 37 assert(DbgID == MD_dbg && "dbg kind id drifted"); (void)DbgID; 38 39 // Create the 'tbaa' metadata kind. 40 unsigned TBAAID = getMDKindID("tbaa"); 41 assert(TBAAID == MD_tbaa && "tbaa kind id drifted"); (void)TBAAID; 42 43 // Create the 'prof' metadata kind. 44 unsigned ProfID = getMDKindID("prof"); 45 assert(ProfID == MD_prof && "prof kind id drifted"); (void)ProfID; 46 47 // Create the 'fpmath' metadata kind. 48 unsigned FPAccuracyID = getMDKindID("fpmath"); 49 assert(FPAccuracyID == MD_fpmath && "fpmath kind id drifted"); 50 (void)FPAccuracyID; 51 52 // Create the 'range' metadata kind. 53 unsigned RangeID = getMDKindID("range"); 54 assert(RangeID == MD_range && "range kind id drifted"); 55 (void)RangeID; 56 57 // Create the 'tbaa.struct' metadata kind. 58 unsigned TBAAStructID = getMDKindID("tbaa.struct"); 59 assert(TBAAStructID == MD_tbaa_struct && "tbaa.struct kind id drifted"); 60 (void)TBAAStructID; 61} 62LLVMContext::~LLVMContext() { delete pImpl; } 63 64void LLVMContext::addModule(Module *M) { 65 pImpl->OwnedModules.insert(M); 66} 67 68void LLVMContext::removeModule(Module *M) { 69 pImpl->OwnedModules.erase(M); 70} 71 72//===----------------------------------------------------------------------===// 73// Recoverable Backend Errors 74//===----------------------------------------------------------------------===// 75 76void LLVMContext:: 77setInlineAsmDiagnosticHandler(InlineAsmDiagHandlerTy DiagHandler, 78 void *DiagContext) { 79 pImpl->InlineAsmDiagHandler = DiagHandler; 80 pImpl->InlineAsmDiagContext = DiagContext; 81} 82 83/// getInlineAsmDiagnosticHandler - Return the diagnostic handler set by 84/// setInlineAsmDiagnosticHandler. 85LLVMContext::InlineAsmDiagHandlerTy 86LLVMContext::getInlineAsmDiagnosticHandler() const { 87 return pImpl->InlineAsmDiagHandler; 88} 89 90/// getInlineAsmDiagnosticContext - Return the diagnostic context set by 91/// setInlineAsmDiagnosticHandler. 92void *LLVMContext::getInlineAsmDiagnosticContext() const { 93 return pImpl->InlineAsmDiagContext; 94} 95 96void LLVMContext::emitError(const Twine &ErrorStr) { 97 emitError(0U, ErrorStr); 98} 99 100void LLVMContext::emitError(const Instruction *I, const Twine &ErrorStr) { 101 unsigned LocCookie = 0; 102 if (const MDNode *SrcLoc = I->getMetadata("srcloc")) { 103 if (SrcLoc->getNumOperands() != 0) 104 if (const ConstantInt *CI = dyn_cast<ConstantInt>(SrcLoc->getOperand(0))) 105 LocCookie = CI->getZExtValue(); 106 } 107 return emitError(LocCookie, ErrorStr); 108} 109 110void LLVMContext::emitError(unsigned LocCookie, const Twine &ErrorStr) { 111 // If there is no error handler installed, just print the error and exit. 112 if (pImpl->InlineAsmDiagHandler == 0) { 113 errs() << "error: " << ErrorStr << "\n"; 114 exit(1); 115 } 116 117 // If we do have an error handler, we can report the error and keep going. 118 SMDiagnostic Diag("", SourceMgr::DK_Error, ErrorStr.str()); 119 120 pImpl->InlineAsmDiagHandler(Diag, pImpl->InlineAsmDiagContext, LocCookie); 121} 122 123//===----------------------------------------------------------------------===// 124// Metadata Kind Uniquing 125//===----------------------------------------------------------------------===// 126 127#ifndef NDEBUG 128/// isValidName - Return true if Name is a valid custom metadata handler name. 129static bool isValidName(StringRef MDName) { 130 if (MDName.empty()) 131 return false; 132 133 if (!std::isalpha(MDName[0])) 134 return false; 135 136 for (StringRef::iterator I = MDName.begin() + 1, E = MDName.end(); I != E; 137 ++I) { 138 if (!std::isalnum(*I) && *I != '_' && *I != '-' && *I != '.') 139 return false; 140 } 141 return true; 142} 143#endif 144 145/// getMDKindID - Return a unique non-zero ID for the specified metadata kind. 146unsigned LLVMContext::getMDKindID(StringRef Name) const { 147 assert(isValidName(Name) && "Invalid MDNode name"); 148 149 // If this is new, assign it its ID. 150 return 151 pImpl->CustomMDKindNames.GetOrCreateValue( 152 Name, pImpl->CustomMDKindNames.size()).second; 153} 154 155/// getHandlerNames - Populate client supplied smallvector using custome 156/// metadata name and ID. 157void LLVMContext::getMDKindNames(SmallVectorImpl<StringRef> &Names) const { 158 Names.resize(pImpl->CustomMDKindNames.size()); 159 for (StringMap<unsigned>::const_iterator I = pImpl->CustomMDKindNames.begin(), 160 E = pImpl->CustomMDKindNames.end(); I != E; ++I) 161 Names[I->second] = I->first(); 162} 163