lib/IR/DebugInfo.cpp

   1 //===--- DebugInfo.cpp - Debug Information Helper Classes -----------------===//
   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 the helper classes used to build and interpret debug
  11 // information in LLVM IR form.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/IR/DebugInfo.h"
  16 #include "LLVMContextImpl.h"
  17 #include "llvm/ADT/STLExtras.h"
  18 #include "llvm/ADT/SmallPtrSet.h"
  19 #include "llvm/ADT/SmallString.h"
  20 #include "llvm/Analysis/ValueTracking.h"
  21 #include "llvm/IR/Constants.h"
  22 #include "llvm/IR/DIBuilder.h"
  23 #include "llvm/IR/DerivedTypes.h"
  24 #include "llvm/IR/Instructions.h"
  25 #include "llvm/IR/IntrinsicInst.h"
  26 #include "llvm/IR/Intrinsics.h"
  27 #include "llvm/IR/GVMaterializer.h"
  28 #include "llvm/IR/Module.h"
  29 #include "llvm/IR/ValueHandle.h"
  30 #include "llvm/Support/Debug.h"
  31 #include "llvm/Support/Dwarf.h"
  32 #include "llvm/Support/raw_ostream.h"
  33 using namespace llvm;
  34 using namespace llvm::dwarf;
  35
  36 //===----------------------------------------------------------------------===//
  37 // Simple Descriptor Constructors and other Methods
  38 //===----------------------------------------------------------------------===//
  39
  40 DIScopeRef DIScope::getRef() const { return MDScopeRef::get(get()); }
  41
  42 void DICompileUnit::replaceSubprograms(DIArray Subprograms) {
  43   get()->replaceSubprograms(MDSubprogramArray(Subprograms));
  44 }
  45
  46 void DICompileUnit::replaceGlobalVariables(DIArray GlobalVariables) {
  47   get()->replaceGlobalVariables(MDGlobalVariableArray(GlobalVariables));
  48 }
  49
  50 DIVariable llvm::createInlinedVariable(MDNode *DV, MDNode *InlinedScope,
  51                                        LLVMContext &VMContext) {
  52   return cast<MDLocalVariable>(DV)
  53       ->withInline(cast_or_null<MDLocation>(InlinedScope));
  54 }
  55
  56 DIVariable llvm::cleanseInlinedVariable(MDNode *DV, LLVMContext &VMContext) {
  57   return cast<MDLocalVariable>(DV)->withoutInline();
  58 }
  59
  60 DISubprogram llvm::getDISubprogram(const MDNode *Scope) {
  61   if (auto *LocalScope = dyn_cast_or_null<MDLocalScope>(Scope))
  62     return LocalScope->getSubprogram();
  63   return nullptr;
  64 }
  65
  66 DISubprogram llvm::getDISubprogram(const Function *F) {
  67   // We look for the first instr that has a debug annotation leading back to F.
  68   for (auto &BB : *F) {
  69     auto Inst = std::find_if(BB.begin(), BB.end(), [](const Instruction &Inst) {
  70       return Inst.getDebugLoc();
  71     });
  72     if (Inst == BB.end())
  73       continue;
  74     DebugLoc DLoc = Inst->getDebugLoc();
  75     const MDNode *Scope = DLoc.getInlinedAtScope();
  76     DISubprogram Subprogram = getDISubprogram(Scope);
  77     return Subprogram.describes(F) ? Subprogram : DISubprogram();
  78   }
  79
  80   return DISubprogram();
  81 }
  82
  83 DICompositeType llvm::getDICompositeType(DIType T) {
  84   if (auto *C = dyn_cast_or_null<MDCompositeTypeBase>(T))
  85     return C;
  86
  87   if (auto *D = dyn_cast_or_null<MDDerivedTypeBase>(T)) {
  88     // This function is currently used by dragonegg and dragonegg does
  89     // not generate identifier for types, so using an empty map to resolve
  90     // DerivedFrom should be fine.
  91     DITypeIdentifierMap EmptyMap;
  92     return getDICompositeType(
  93         DIDerivedType(D).getTypeDerivedFrom().resolve(EmptyMap));
  94   }
  95
  96   return nullptr;
  97 }
  98
  99 DITypeIdentifierMap
 100 llvm::generateDITypeIdentifierMap(const NamedMDNode *CU_Nodes) {
 101   DITypeIdentifierMap Map;
 102   for (unsigned CUi = 0, CUe = CU_Nodes->getNumOperands(); CUi != CUe; ++CUi) {
 103     DICompileUnit CU = cast<MDCompileUnit>(CU_Nodes->getOperand(CUi));
 104     DIArray Retain = CU.getRetainedTypes();
 105     for (unsigned Ti = 0, Te = Retain.size(); Ti != Te; ++Ti) {
 106       if (!isa<MDCompositeType>(Retain[Ti]))
 107         continue;
 108       DICompositeType Ty = cast<MDCompositeType>(Retain[Ti]);
 109       if (MDString *TypeId = Ty.getIdentifier()) {
 110         // Definition has priority over declaration.
 111         // Try to insert (TypeId, Ty) to Map.
 112         std::pair<DITypeIdentifierMap::iterator, bool> P =
 113             Map.insert(std::make_pair(TypeId, Ty));
 114         // If TypeId already exists in Map and this is a definition, replace
 115         // whatever we had (declaration or definition) with the definition.
 116         if (!P.second && !Ty.isForwardDecl())
 117           P.first->second = Ty;
 118       }
 119     }
 120   }
 121   return Map;
 122 }
 123
 124 //===----------------------------------------------------------------------===//
 125 // DebugInfoFinder implementations.
 126 //===----------------------------------------------------------------------===//
 127
 128 void DebugInfoFinder::reset() {
 129   CUs.clear();
 130   SPs.clear();
 131   GVs.clear();
 132   TYs.clear();
 133   Scopes.clear();
 134   NodesSeen.clear();
 135   TypeIdentifierMap.clear();
 136   TypeMapInitialized = false;
 137 }
 138
 139 void DebugInfoFinder::InitializeTypeMap(const Module &M) {
 140   if (!TypeMapInitialized)
 141     if (NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu")) {
 142       TypeIdentifierMap = generateDITypeIdentifierMap(CU_Nodes);
 143       TypeMapInitialized = true;
 144     }
 145 }
 146
 147 void DebugInfoFinder::processModule(const Module &M) {
 148   InitializeTypeMap(M);
 149   if (NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu")) {
 150     for (unsigned i = 0, e = CU_Nodes->getNumOperands(); i != e; ++i) {
 151       DICompileUnit CU = cast<MDCompileUnit>(CU_Nodes->getOperand(i));
 152       addCompileUnit(CU);
 153       for (DIGlobalVariable DIG : CU->getGlobalVariables()) {
 154         if (addGlobalVariable(DIG)) {
 155           processScope(DIG->getScope());
 156           processType(DIG->getType().resolve(TypeIdentifierMap));
 157         }
 158       }
 159       for (auto *SP : CU->getSubprograms())
 160         processSubprogram(SP);
 161       for (auto *ET : CU->getEnumTypes())
 162         processType(ET);
 163       for (auto *RT : CU->getRetainedTypes())
 164         processType(RT);
 165       for (DIImportedEntity Import : CU->getImportedEntities()) {
 166         auto *Entity = Import->getEntity().resolve(TypeIdentifierMap);
 167         if (auto *T = dyn_cast<MDType>(Entity))
 168           processType(T);
 169         else if (auto *SP = dyn_cast<MDSubprogram>(Entity))
 170           processSubprogram(SP);
 171         else if (auto *NS = dyn_cast<MDNamespace>(Entity))
 172           processScope(NS->getScope());
 173       }
 174     }
 175   }
 176 }
 177
 178 void DebugInfoFinder::processLocation(const Module &M, DILocation Loc) {
 179   if (!Loc)
 180     return;
 181   InitializeTypeMap(M);
 182   processScope(Loc->getScope());
 183   processLocation(M, Loc->getInlinedAt());
 184 }
 185
 186 void DebugInfoFinder::processType(DIType DT) {
 187   if (!addType(DT))
 188     return;
 189   processScope(DT.getContext().resolve(TypeIdentifierMap));
 190   if (DICompositeType DCT = dyn_cast<MDCompositeTypeBase>(DT)) {
 191     processType(DCT.getTypeDerivedFrom().resolve(TypeIdentifierMap));
 192     if (DISubroutineType ST = dyn_cast<MDSubroutineType>(DCT)) {
 193       for (MDTypeRef Ref : ST->getTypeArray())
 194         processType(Ref.resolve(TypeIdentifierMap));
 195       return;
 196     }
 197     for (Metadata *D : DCT->getElements()->operands()) {
 198       if (DIType T = dyn_cast<MDType>(D))
 199         processType(T);
 200       else if (DISubprogram SP = dyn_cast<MDSubprogram>(D))
 201         processSubprogram(SP);
 202     }
 203   } else if (DIDerivedType DDT = dyn_cast<MDDerivedTypeBase>(DT)) {
 204     processType(DDT.getTypeDerivedFrom().resolve(TypeIdentifierMap));
 205   }
 206 }
 207
 208 void DebugInfoFinder::processScope(DIScope Scope) {
 209   if (!Scope)
 210     return;
 211   if (DIType Ty = dyn_cast<MDType>(Scope)) {
 212     processType(Ty);
 213     return;
 214   }
 215   if (DICompileUnit CU = dyn_cast<MDCompileUnit>(Scope)) {
 216     addCompileUnit(CU);
 217     return;
 218   }
 219   if (DISubprogram SP = dyn_cast<MDSubprogram>(Scope)) {
 220     processSubprogram(SP);
 221     return;
 222   }
 223   if (!addScope(Scope))
 224     return;
 225   if (DILexicalBlock LB = dyn_cast<MDLexicalBlockBase>(Scope)) {
 226     processScope(LB.getContext());
 227   } else if (DINameSpace NS = dyn_cast<MDNamespace>(Scope)) {
 228     processScope(NS.getContext());
 229   }
 230 }
 231
 232 void DebugInfoFinder::processSubprogram(DISubprogram SP) {
 233   if (!addSubprogram(SP))
 234     return;
 235   processScope(SP.getContext().resolve(TypeIdentifierMap));
 236   processType(SP.getType());
 237   for (auto *Element : SP.getTemplateParams()) {
 238     if (DITemplateTypeParameter TType =
 239             dyn_cast<MDTemplateTypeParameter>(Element)) {
 240       processType(TType.getType().resolve(TypeIdentifierMap));
 241     } else if (DITemplateValueParameter TVal =
 242                    dyn_cast<MDTemplateValueParameter>(Element)) {
 243       processType(TVal.getType().resolve(TypeIdentifierMap));
 244     }
 245   }
 246 }
 247
 248 void DebugInfoFinder::processDeclare(const Module &M,
 249                                      const DbgDeclareInst *DDI) {
 250   MDNode *N = dyn_cast<MDNode>(DDI->getVariable());
 251   if (!N)
 252     return;
 253   InitializeTypeMap(M);
 254
 255   DIVariable DV = dyn_cast<MDLocalVariable>(N);
 256   if (!DV)
 257     return;
 258
 259   if (!NodesSeen.insert(DV).second)
 260     return;
 261   processScope(DV->getScope());
 262   processType(DV->getType().resolve(TypeIdentifierMap));
 263 }
 264
 265 void DebugInfoFinder::processValue(const Module &M, const DbgValueInst *DVI) {
 266   MDNode *N = dyn_cast<MDNode>(DVI->getVariable());
 267   if (!N)
 268     return;
 269   InitializeTypeMap(M);
 270
 271   DIVariable DV = dyn_cast<MDLocalVariable>(N);
 272   if (!DV)
 273     return;
 274
 275   if (!NodesSeen.insert(DV).second)
 276     return;
 277   processScope(DV->getScope());
 278   processType(DV->getType().resolve(TypeIdentifierMap));
 279 }
 280
 281 bool DebugInfoFinder::addType(DIType DT) {
 282   if (!DT)
 283     return false;
 284
 285   if (!NodesSeen.insert(DT).second)
 286     return false;
 287
 288   TYs.push_back(DT);
 289   return true;
 290 }
 291
 292 bool DebugInfoFinder::addCompileUnit(DICompileUnit CU) {
 293   if (!CU)
 294     return false;
 295   if (!NodesSeen.insert(CU).second)
 296     return false;
 297
 298   CUs.push_back(CU);
 299   return true;
 300 }
 301
 302 bool DebugInfoFinder::addGlobalVariable(DIGlobalVariable DIG) {
 303   if (!DIG)
 304     return false;
 305
 306   if (!NodesSeen.insert(DIG).second)
 307     return false;
 308
 309   GVs.push_back(DIG);
 310   return true;
 311 }
 312
 313 bool DebugInfoFinder::addSubprogram(DISubprogram SP) {
 314   if (!SP)
 315     return false;
 316
 317   if (!NodesSeen.insert(SP).second)
 318     return false;
 319
 320   SPs.push_back(SP);
 321   return true;
 322 }
 323
 324 bool DebugInfoFinder::addScope(DIScope Scope) {
 325   if (!Scope)
 326     return false;
 327   // FIXME: Ocaml binding generates a scope with no content, we treat it
 328   // as null for now.
 329   if (Scope->getNumOperands() == 0)
 330     return false;
 331   if (!NodesSeen.insert(Scope).second)
 332     return false;
 333   Scopes.push_back(Scope);
 334   return true;
 335 }
 336
 337 //===----------------------------------------------------------------------===//
 338 // DIDescriptor: dump routines for all descriptors.
 339 //===----------------------------------------------------------------------===//
 340
 341 void DIDescriptor::dump() const {
 342   print(dbgs());
 343   dbgs() << '\n';
 344 }
 345
 346 void DIDescriptor::print(raw_ostream &OS) const {
 347   if (!get())
 348     return;
 349   get()->print(OS);
 350 }
 351
 352 template <>
 353 DIDescriptor
 354 DIRef<DIDescriptor>::resolve(const DITypeIdentifierMap &Map) const {
 355   return DIDescriptor(DebugNodeRef(Val).resolve(Map));
 356 }
 357 template <>
 358 DIScope DIRef<DIScope>::resolve(const DITypeIdentifierMap &Map) const {
 359   return MDScopeRef(Val).resolve(Map);
 360 }
 361 template <>
 362 DIType DIRef<DIType>::resolve(const DITypeIdentifierMap &Map) const {
 363   return MDTypeRef(Val).resolve(Map);
 364 }
 365
 366 bool llvm::stripDebugInfo(Function &F) {
 367   bool Changed = false;
 368   for (BasicBlock &BB : F) {
 369     for (Instruction &I : BB) {
 370       if (I.getDebugLoc()) {
 371         Changed = true;
 372         I.setDebugLoc(DebugLoc());
 373       }
 374     }
 375   }
 376   return Changed;
 377 }
 378
 379 bool llvm::StripDebugInfo(Module &M) {
 380   bool Changed = false;
 381
 382   // Remove all of the calls to the debugger intrinsics, and remove them from
 383   // the module.
 384   if (Function *Declare = M.getFunction("llvm.dbg.declare")) {
 385     while (!Declare->use_empty()) {
 386       CallInst *CI = cast<CallInst>(Declare->user_back());
 387       CI->eraseFromParent();
 388     }
 389     Declare->eraseFromParent();
 390     Changed = true;
 391   }
 392
 393   if (Function *DbgVal = M.getFunction("llvm.dbg.value")) {
 394     while (!DbgVal->use_empty()) {
 395       CallInst *CI = cast<CallInst>(DbgVal->user_back());
 396       CI->eraseFromParent();
 397     }
 398     DbgVal->eraseFromParent();
 399     Changed = true;
 400   }
 401
 402   for (Module::named_metadata_iterator NMI = M.named_metadata_begin(),
 403          NME = M.named_metadata_end(); NMI != NME;) {
 404     NamedMDNode *NMD = NMI;
 405     ++NMI;
 406     if (NMD->getName().startswith("llvm.dbg.")) {
 407       NMD->eraseFromParent();
 408       Changed = true;
 409     }
 410   }
 411
 412   for (Function &F : M)
 413     Changed |= stripDebugInfo(F);
 414
 415   if (GVMaterializer *Materializer = M.getMaterializer())
 416     Materializer->setStripDebugInfo();
 417
 418   return Changed;
 419 }
 420
 421 unsigned llvm::getDebugMetadataVersionFromModule(const Module &M) {
 422   if (auto *Val = mdconst::dyn_extract_or_null<ConstantInt>(
 423           M.getModuleFlag("Debug Info Version")))
 424     return Val->getZExtValue();
 425   return 0;
 426 }
 427
 428 llvm::DenseMap<const llvm::Function *, llvm::DISubprogram>
 429 llvm::makeSubprogramMap(const Module &M) {
 430   DenseMap<const Function *, DISubprogram> R;
 431
 432   NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu");
 433   if (!CU_Nodes)
 434     return R;
 435
 436   for (MDNode *N : CU_Nodes->operands()) {
 437     DICompileUnit CUNode = cast<MDCompileUnit>(N);
 438     for (DISubprogram SP : CUNode->getSubprograms()) {
 439       if (Function *F = SP.getFunction())
 440         R.insert(std::make_pair(F, SP));
 441     }
 442   }
 443   return R;
 444 }