Home | History | Annotate | Download | only in VMCore
      1 //===-- DebugLoc.cpp - Implement DebugLoc class ---------------------------===//
      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 #include "llvm/Support/DebugLoc.h"
     11 #include "llvm/ADT/DenseMapInfo.h"
     12 #include "LLVMContextImpl.h"
     13 using namespace llvm;
     14 
     15 //===----------------------------------------------------------------------===//
     16 // DebugLoc Implementation
     17 //===----------------------------------------------------------------------===//
     18 
     19 MDNode *DebugLoc::getScope(const LLVMContext &Ctx) const {
     20   if (ScopeIdx == 0) return 0;
     21 
     22   if (ScopeIdx > 0) {
     23     // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
     24     // position specified.
     25     assert(unsigned(ScopeIdx) <= Ctx.pImpl->ScopeRecords.size() &&
     26            "Invalid ScopeIdx!");
     27     return Ctx.pImpl->ScopeRecords[ScopeIdx-1].get();
     28   }
     29 
     30   // Otherwise, the index is in the ScopeInlinedAtRecords array.
     31   assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
     32          "Invalid ScopeIdx");
     33   return Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].first.get();
     34 }
     35 
     36 MDNode *DebugLoc::getInlinedAt(const LLVMContext &Ctx) const {
     37   // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
     38   // position specified.  Zero is invalid.
     39   if (ScopeIdx >= 0) return 0;
     40 
     41   // Otherwise, the index is in the ScopeInlinedAtRecords array.
     42   assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
     43          "Invalid ScopeIdx");
     44   return Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].second.get();
     45 }
     46 
     47 /// Return both the Scope and the InlinedAt values.
     48 void DebugLoc::getScopeAndInlinedAt(MDNode *&Scope, MDNode *&IA,
     49                                     const LLVMContext &Ctx) const {
     50   if (ScopeIdx == 0) {
     51     Scope = IA = 0;
     52     return;
     53   }
     54 
     55   if (ScopeIdx > 0) {
     56     // Positive ScopeIdx is an index into ScopeRecords, which has no inlined-at
     57     // position specified.
     58     assert(unsigned(ScopeIdx) <= Ctx.pImpl->ScopeRecords.size() &&
     59            "Invalid ScopeIdx!");
     60     Scope = Ctx.pImpl->ScopeRecords[ScopeIdx-1].get();
     61     IA = 0;
     62     return;
     63   }
     64 
     65   // Otherwise, the index is in the ScopeInlinedAtRecords array.
     66   assert(unsigned(-ScopeIdx) <= Ctx.pImpl->ScopeInlinedAtRecords.size() &&
     67          "Invalid ScopeIdx");
     68   Scope = Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].first.get();
     69   IA    = Ctx.pImpl->ScopeInlinedAtRecords[-ScopeIdx-1].second.get();
     70 }
     71 
     72 
     73 DebugLoc DebugLoc::get(unsigned Line, unsigned Col,
     74                        MDNode *Scope, MDNode *InlinedAt) {
     75   DebugLoc Result;
     76 
     77   // If no scope is available, this is an unknown location.
     78   if (Scope == 0) return Result;
     79 
     80   // Saturate line and col to "unknown".
     81   if (Col > 255) Col = 0;
     82   if (Line >= (1 << 24)) Line = 0;
     83   Result.LineCol = Line | (Col << 24);
     84 
     85   LLVMContext &Ctx = Scope->getContext();
     86 
     87   // If there is no inlined-at location, use the ScopeRecords array.
     88   if (InlinedAt == 0)
     89     Result.ScopeIdx = Ctx.pImpl->getOrAddScopeRecordIdxEntry(Scope, 0);
     90   else
     91     Result.ScopeIdx = Ctx.pImpl->getOrAddScopeInlinedAtIdxEntry(Scope,
     92                                                                 InlinedAt, 0);
     93 
     94   return Result;
     95 }
     96 
     97 /// getAsMDNode - This method converts the compressed DebugLoc node into a
     98 /// DILocation compatible MDNode.
     99 MDNode *DebugLoc::getAsMDNode(const LLVMContext &Ctx) const {
    100   if (isUnknown()) return 0;
    101 
    102   MDNode *Scope, *IA;
    103   getScopeAndInlinedAt(Scope, IA, Ctx);
    104   assert(Scope && "If scope is null, this should be isUnknown()");
    105 
    106   LLVMContext &Ctx2 = Scope->getContext();
    107   Type *Int32 = Type::getInt32Ty(Ctx2);
    108   Value *Elts[] = {
    109     ConstantInt::get(Int32, getLine()), ConstantInt::get(Int32, getCol()),
    110     Scope, IA
    111   };
    112   return MDNode::get(Ctx2, Elts);
    113 }
    114 
    115 /// getFromDILocation - Translate the DILocation quad into a DebugLoc.
    116 DebugLoc DebugLoc::getFromDILocation(MDNode *N) {
    117   if (N == 0 || N->getNumOperands() != 4) return DebugLoc();
    118 
    119   MDNode *Scope = dyn_cast_or_null<MDNode>(N->getOperand(2));
    120   if (Scope == 0) return DebugLoc();
    121 
    122   unsigned LineNo = 0, ColNo = 0;
    123   if (ConstantInt *Line = dyn_cast_or_null<ConstantInt>(N->getOperand(0)))
    124     LineNo = Line->getZExtValue();
    125   if (ConstantInt *Col = dyn_cast_or_null<ConstantInt>(N->getOperand(1)))
    126     ColNo = Col->getZExtValue();
    127 
    128   return get(LineNo, ColNo, Scope, dyn_cast_or_null<MDNode>(N->getOperand(3)));
    129 }
    130 
    131 /// getFromDILexicalBlock - Translate the DILexicalBlock into a DebugLoc.
    132 DebugLoc DebugLoc::getFromDILexicalBlock(MDNode *N) {
    133   if (N == 0 || N->getNumOperands() < 3) return DebugLoc();
    134 
    135   MDNode *Scope = dyn_cast_or_null<MDNode>(N->getOperand(1));
    136   if (Scope == 0) return DebugLoc();
    137 
    138   unsigned LineNo = 0, ColNo = 0;
    139   if (ConstantInt *Line = dyn_cast_or_null<ConstantInt>(N->getOperand(2)))
    140     LineNo = Line->getZExtValue();
    141   if (ConstantInt *Col = dyn_cast_or_null<ConstantInt>(N->getOperand(3)))
    142     ColNo = Col->getZExtValue();
    143 
    144   return get(LineNo, ColNo, Scope, NULL);
    145 }
    146 
    147 void DebugLoc::dump(const LLVMContext &Ctx) const {
    148 #ifndef NDEBUG
    149   if (!isUnknown()) {
    150     dbgs() << getLine();
    151     if (getCol() != 0)
    152       dbgs() << ',' << getCol();
    153     DebugLoc InlinedAtDL = DebugLoc::getFromDILocation(getInlinedAt(Ctx));
    154     if (!InlinedAtDL.isUnknown()) {
    155       dbgs() << " @ ";
    156       InlinedAtDL.dump(Ctx);
    157     } else
    158       dbgs() << "\n";
    159   }
    160 #endif
    161 }
    162 
    163 //===----------------------------------------------------------------------===//
    164 // DenseMap specialization
    165 //===----------------------------------------------------------------------===//
    166 
    167 DebugLoc DenseMapInfo<DebugLoc>::getEmptyKey() {
    168   return DebugLoc::getEmptyKey();
    169 }
    170 
    171 DebugLoc DenseMapInfo<DebugLoc>::getTombstoneKey() {
    172   return DebugLoc::getTombstoneKey();
    173 }
    174 
    175 unsigned DenseMapInfo<DebugLoc>::getHashValue(const DebugLoc &Key) {
    176   FoldingSetNodeID ID;
    177   ID.AddInteger(Key.LineCol);
    178   ID.AddInteger(Key.ScopeIdx);
    179   return ID.ComputeHash();
    180 }
    181 
    182 bool DenseMapInfo<DebugLoc>::isEqual(const DebugLoc &LHS, const DebugLoc &RHS) {
    183   return LHS == RHS;
    184 }
    185 
    186 //===----------------------------------------------------------------------===//
    187 // LLVMContextImpl Implementation
    188 //===----------------------------------------------------------------------===//
    189 
    190 int LLVMContextImpl::getOrAddScopeRecordIdxEntry(MDNode *Scope,
    191                                                  int ExistingIdx) {
    192   // If we already have an entry for this scope, return it.
    193   int &Idx = ScopeRecordIdx[Scope];
    194   if (Idx) return Idx;
    195 
    196   // If we don't have an entry, but ExistingIdx is specified, use it.
    197   if (ExistingIdx)
    198     return Idx = ExistingIdx;
    199 
    200   // Otherwise add a new entry.
    201 
    202   // Start out ScopeRecords with a minimal reasonable size to avoid
    203   // excessive reallocation starting out.
    204   if (ScopeRecords.empty())
    205     ScopeRecords.reserve(128);
    206 
    207   // Index is biased by 1 for index.
    208   Idx = ScopeRecords.size()+1;
    209   ScopeRecords.push_back(DebugRecVH(Scope, this, Idx));
    210   return Idx;
    211 }
    212 
    213 int LLVMContextImpl::getOrAddScopeInlinedAtIdxEntry(MDNode *Scope, MDNode *IA,
    214                                                     int ExistingIdx) {
    215   // If we already have an entry, return it.
    216   int &Idx = ScopeInlinedAtIdx[std::make_pair(Scope, IA)];
    217   if (Idx) return Idx;
    218 
    219   // If we don't have an entry, but ExistingIdx is specified, use it.
    220   if (ExistingIdx)
    221     return Idx = ExistingIdx;
    222 
    223   // Start out ScopeInlinedAtRecords with a minimal reasonable size to avoid
    224   // excessive reallocation starting out.
    225   if (ScopeInlinedAtRecords.empty())
    226     ScopeInlinedAtRecords.reserve(128);
    227 
    228   // Index is biased by 1 and negated.
    229   Idx = -ScopeInlinedAtRecords.size()-1;
    230   ScopeInlinedAtRecords.push_back(std::make_pair(DebugRecVH(Scope, this, Idx),
    231                                                  DebugRecVH(IA, this, Idx)));
    232   return Idx;
    233 }
    234 
    235 
    236 //===----------------------------------------------------------------------===//
    237 // DebugRecVH Implementation
    238 //===----------------------------------------------------------------------===//
    239 
    240 /// deleted - The MDNode this is pointing to got deleted, so this pointer needs
    241 /// to drop to null and we need remove our entry from the DenseMap.
    242 void DebugRecVH::deleted() {
    243   // If this is a non-canonical reference, just drop the value to null, we know
    244   // it doesn't have a map entry.
    245   if (Idx == 0) {
    246     setValPtr(0);
    247     return;
    248   }
    249 
    250   MDNode *Cur = get();
    251 
    252   // If the index is positive, it is an entry in ScopeRecords.
    253   if (Idx > 0) {
    254     assert(Ctx->ScopeRecordIdx[Cur] == Idx && "Mapping out of date!");
    255     Ctx->ScopeRecordIdx.erase(Cur);
    256     // Reset this VH to null and we're done.
    257     setValPtr(0);
    258     Idx = 0;
    259     return;
    260   }
    261 
    262   // Otherwise, it is an entry in ScopeInlinedAtRecords, we don't know if it
    263   // is the scope or the inlined-at record entry.
    264   assert(unsigned(-Idx-1) < Ctx->ScopeInlinedAtRecords.size());
    265   std::pair<DebugRecVH, DebugRecVH> &Entry = Ctx->ScopeInlinedAtRecords[-Idx-1];
    266   assert((this == &Entry.first || this == &Entry.second) &&
    267          "Mapping out of date!");
    268 
    269   MDNode *OldScope = Entry.first.get();
    270   MDNode *OldInlinedAt = Entry.second.get();
    271   assert(OldScope != 0 && OldInlinedAt != 0 &&
    272          "Entry should be non-canonical if either val dropped to null");
    273 
    274   // Otherwise, we do have an entry in it, nuke it and we're done.
    275   assert(Ctx->ScopeInlinedAtIdx[std::make_pair(OldScope, OldInlinedAt)] == Idx&&
    276          "Mapping out of date");
    277   Ctx->ScopeInlinedAtIdx.erase(std::make_pair(OldScope, OldInlinedAt));
    278 
    279   // Reset this VH to null.  Drop both 'Idx' values to null to indicate that
    280   // we're in non-canonical form now.
    281   setValPtr(0);
    282   Entry.first.Idx = Entry.second.Idx = 0;
    283 }
    284 
    285 void DebugRecVH::allUsesReplacedWith(Value *NewVa) {
    286   // If being replaced with a non-mdnode value (e.g. undef) handle this as if
    287   // the mdnode got deleted.
    288   MDNode *NewVal = dyn_cast<MDNode>(NewVa);
    289   if (NewVal == 0) return deleted();
    290 
    291   // If this is a non-canonical reference, just change it, we know it already
    292   // doesn't have a map entry.
    293   if (Idx == 0) {
    294     setValPtr(NewVa);
    295     return;
    296   }
    297 
    298   MDNode *OldVal = get();
    299   assert(OldVal != NewVa && "Node replaced with self?");
    300 
    301   // If the index is positive, it is an entry in ScopeRecords.
    302   if (Idx > 0) {
    303     assert(Ctx->ScopeRecordIdx[OldVal] == Idx && "Mapping out of date!");
    304     Ctx->ScopeRecordIdx.erase(OldVal);
    305     setValPtr(NewVal);
    306 
    307     int NewEntry = Ctx->getOrAddScopeRecordIdxEntry(NewVal, Idx);
    308 
    309     // If NewVal already has an entry, this becomes a non-canonical reference,
    310     // just drop Idx to 0 to signify this.
    311     if (NewEntry != Idx)
    312       Idx = 0;
    313     return;
    314   }
    315 
    316   // Otherwise, it is an entry in ScopeInlinedAtRecords, we don't know if it
    317   // is the scope or the inlined-at record entry.
    318   assert(unsigned(-Idx-1) < Ctx->ScopeInlinedAtRecords.size());
    319   std::pair<DebugRecVH, DebugRecVH> &Entry = Ctx->ScopeInlinedAtRecords[-Idx-1];
    320   assert((this == &Entry.first || this == &Entry.second) &&
    321          "Mapping out of date!");
    322 
    323   MDNode *OldScope = Entry.first.get();
    324   MDNode *OldInlinedAt = Entry.second.get();
    325   assert(OldScope != 0 && OldInlinedAt != 0 &&
    326          "Entry should be non-canonical if either val dropped to null");
    327 
    328   // Otherwise, we do have an entry in it, nuke it and we're done.
    329   assert(Ctx->ScopeInlinedAtIdx[std::make_pair(OldScope, OldInlinedAt)] == Idx&&
    330          "Mapping out of date");
    331   Ctx->ScopeInlinedAtIdx.erase(std::make_pair(OldScope, OldInlinedAt));
    332 
    333   // Reset this VH to the new value.
    334   setValPtr(NewVal);
    335 
    336   int NewIdx = Ctx->getOrAddScopeInlinedAtIdxEntry(Entry.first.get(),
    337                                                    Entry.second.get(), Idx);
    338   // If NewVal already has an entry, this becomes a non-canonical reference,
    339   // just drop Idx to 0 to signify this.
    340   if (NewIdx != Idx) {
    341     std::pair<DebugRecVH, DebugRecVH> &Entry=Ctx->ScopeInlinedAtRecords[-Idx-1];
    342     Entry.first.Idx = Entry.second.Idx = 0;
    343   }
    344 }
    345