xref: /llvm-project/llvm/lib/CodeGen/LiveDebugVariables.cpp (revision 1368434b4973918b04b406633ce28d4e41481e8d)
1 //===- LiveDebugVariables.cpp - Tracking debug info variables -------------===//
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 LiveDebugVariables analysis.
11 //
12 // Remove all DBG_VALUE instructions referencing virtual registers and replace
13 // them with a data structure tracking where live user variables are kept - in a
14 // virtual register or in a stack slot.
15 //
16 // Allow the data structure to be updated during register allocation when values
17 // are moved between registers and stack slots. Finally emit new DBG_VALUE
18 // instructions after register allocation is complete.
19 //
20 //===----------------------------------------------------------------------===//
21 
22 #include "LiveDebugVariables.h"
23 #include "llvm/ADT/ArrayRef.h"
24 #include "llvm/ADT/DenseMap.h"
25 #include "llvm/ADT/IntervalMap.h"
26 #include "llvm/ADT/STLExtras.h"
27 #include "llvm/ADT/SmallSet.h"
28 #include "llvm/ADT/SmallVector.h"
29 #include "llvm/ADT/Statistic.h"
30 #include "llvm/ADT/StringRef.h"
31 #include "llvm/CodeGen/LexicalScopes.h"
32 #include "llvm/CodeGen/LiveInterval.h"
33 #include "llvm/CodeGen/LiveIntervals.h"
34 #include "llvm/CodeGen/MachineBasicBlock.h"
35 #include "llvm/CodeGen/MachineDominators.h"
36 #include "llvm/CodeGen/MachineFunction.h"
37 #include "llvm/CodeGen/MachineInstr.h"
38 #include "llvm/CodeGen/MachineInstrBuilder.h"
39 #include "llvm/CodeGen/MachineOperand.h"
40 #include "llvm/CodeGen/MachineRegisterInfo.h"
41 #include "llvm/CodeGen/SlotIndexes.h"
42 #include "llvm/CodeGen/TargetInstrInfo.h"
43 #include "llvm/CodeGen/TargetOpcodes.h"
44 #include "llvm/CodeGen/TargetRegisterInfo.h"
45 #include "llvm/CodeGen/TargetSubtargetInfo.h"
46 #include "llvm/CodeGen/VirtRegMap.h"
47 #include "llvm/Config/llvm-config.h"
48 #include "llvm/IR/DebugInfoMetadata.h"
49 #include "llvm/IR/DebugLoc.h"
50 #include "llvm/IR/Function.h"
51 #include "llvm/IR/Metadata.h"
52 #include "llvm/MC/MCRegisterInfo.h"
53 #include "llvm/Pass.h"
54 #include "llvm/Support/Casting.h"
55 #include "llvm/Support/CommandLine.h"
56 #include "llvm/Support/Compiler.h"
57 #include "llvm/Support/Debug.h"
58 #include "llvm/Support/raw_ostream.h"
59 #include <algorithm>
60 #include <cassert>
61 #include <iterator>
62 #include <memory>
63 #include <utility>
64 
65 using namespace llvm;
66 
67 #define DEBUG_TYPE "livedebugvars"
68 
69 static cl::opt<bool>
70 EnableLDV("live-debug-variables", cl::init(true),
71           cl::desc("Enable the live debug variables pass"), cl::Hidden);
72 
73 STATISTIC(NumInsertedDebugValues, "Number of DBG_VALUEs inserted");
74 
75 char LiveDebugVariables::ID = 0;
76 
77 INITIALIZE_PASS_BEGIN(LiveDebugVariables, DEBUG_TYPE,
78                 "Debug Variable Analysis", false, false)
79 INITIALIZE_PASS_DEPENDENCY(MachineDominatorTree)
80 INITIALIZE_PASS_DEPENDENCY(LiveIntervals)
81 INITIALIZE_PASS_END(LiveDebugVariables, DEBUG_TYPE,
82                 "Debug Variable Analysis", false, false)
83 
84 void LiveDebugVariables::getAnalysisUsage(AnalysisUsage &AU) const {
85   AU.addRequired<MachineDominatorTree>();
86   AU.addRequiredTransitive<LiveIntervals>();
87   AU.setPreservesAll();
88   MachineFunctionPass::getAnalysisUsage(AU);
89 }
90 
91 LiveDebugVariables::LiveDebugVariables() : MachineFunctionPass(ID) {
92   initializeLiveDebugVariablesPass(*PassRegistry::getPassRegistry());
93 }
94 
95 enum : unsigned { UndefLocNo = ~0U };
96 
97 /// Describes a location by number along with some flags about the original
98 /// usage of the location.
99 class DbgValueLocation {
100 public:
101   DbgValueLocation(unsigned LocNo, bool WasIndirect)
102       : LocNo(LocNo), WasIndirect(WasIndirect) {
103     static_assert(sizeof(*this) == sizeof(unsigned), "bad bitfield packing");
104     assert(locNo() == LocNo && "location truncation");
105   }
106 
107   DbgValueLocation() : LocNo(0), WasIndirect(0) {}
108 
109   unsigned locNo() const {
110     // Fix up the undef location number, which gets truncated.
111     return LocNo == INT_MAX ? UndefLocNo : LocNo;
112   }
113   bool wasIndirect() const { return WasIndirect; }
114   bool isUndef() const { return locNo() == UndefLocNo; }
115 
116   DbgValueLocation changeLocNo(unsigned NewLocNo) const {
117     return DbgValueLocation(NewLocNo, WasIndirect);
118   }
119 
120   friend inline bool operator==(const DbgValueLocation &LHS,
121                                 const DbgValueLocation &RHS) {
122     return LHS.LocNo == RHS.LocNo && LHS.WasIndirect == RHS.WasIndirect;
123   }
124 
125   friend inline bool operator!=(const DbgValueLocation &LHS,
126                                 const DbgValueLocation &RHS) {
127     return !(LHS == RHS);
128   }
129 
130 private:
131   unsigned LocNo : 31;
132   unsigned WasIndirect : 1;
133 };
134 
135 /// LocMap - Map of where a user value is live, and its location.
136 using LocMap = IntervalMap<SlotIndex, DbgValueLocation, 4>;
137 
138 namespace {
139 
140 class LDVImpl;
141 
142 /// UserValue - A user value is a part of a debug info user variable.
143 ///
144 /// A DBG_VALUE instruction notes that (a sub-register of) a virtual register
145 /// holds part of a user variable. The part is identified by a byte offset.
146 ///
147 /// UserValues are grouped into equivalence classes for easier searching. Two
148 /// user values are related if they refer to the same variable, or if they are
149 /// held by the same virtual register. The equivalence class is the transitive
150 /// closure of that relation.
151 class UserValue {
152   const DILocalVariable *Variable; ///< The debug info variable we are part of.
153   const DIExpression *Expression; ///< Any complex address expression.
154   DebugLoc dl;            ///< The debug location for the variable. This is
155                           ///< used by dwarf writer to find lexical scope.
156   UserValue *leader;      ///< Equivalence class leader.
157   UserValue *next = nullptr; ///< Next value in equivalence class, or null.
158 
159   /// Numbered locations referenced by locmap.
160   SmallVector<MachineOperand, 4> locations;
161 
162   /// Map of slot indices where this value is live.
163   LocMap locInts;
164 
165   /// Set of interval start indexes that have been trimmed to the
166   /// lexical scope.
167   SmallSet<SlotIndex, 2> trimmedDefs;
168 
169   /// insertDebugValue - Insert a DBG_VALUE into MBB at Idx for LocNo.
170   void insertDebugValue(MachineBasicBlock *MBB, SlotIndex StartIdx,
171                         SlotIndex StopIdx,
172                         DbgValueLocation Loc, bool Spilled, LiveIntervals &LIS,
173                         const TargetInstrInfo &TII,
174                         const TargetRegisterInfo &TRI);
175 
176   /// splitLocation - Replace OldLocNo ranges with NewRegs ranges where NewRegs
177   /// is live. Returns true if any changes were made.
178   bool splitLocation(unsigned OldLocNo, ArrayRef<unsigned> NewRegs,
179                      LiveIntervals &LIS);
180 
181 public:
182   /// UserValue - Create a new UserValue.
183   UserValue(const DILocalVariable *var, const DIExpression *expr, DebugLoc L,
184             LocMap::Allocator &alloc)
185       : Variable(var), Expression(expr), dl(std::move(L)), leader(this),
186         locInts(alloc) {}
187 
188   /// getLeader - Get the leader of this value's equivalence class.
189   UserValue *getLeader() {
190     UserValue *l = leader;
191     while (l != l->leader)
192       l = l->leader;
193     return leader = l;
194   }
195 
196   /// getNext - Return the next UserValue in the equivalence class.
197   UserValue *getNext() const { return next; }
198 
199   /// match - Does this UserValue match the parameters?
200   bool match(const DILocalVariable *Var, const DIExpression *Expr,
201              const DILocation *IA) const {
202     // FIXME: The fragment should be part of the equivalence class, but not
203     // other things in the expression like stack values.
204     return Var == Variable && Expr == Expression && dl->getInlinedAt() == IA;
205   }
206 
207   /// merge - Merge equivalence classes.
208   static UserValue *merge(UserValue *L1, UserValue *L2) {
209     L2 = L2->getLeader();
210     if (!L1)
211       return L2;
212     L1 = L1->getLeader();
213     if (L1 == L2)
214       return L1;
215     // Splice L2 before L1's members.
216     UserValue *End = L2;
217     while (End->next) {
218       End->leader = L1;
219       End = End->next;
220     }
221     End->leader = L1;
222     End->next = L1->next;
223     L1->next = L2;
224     return L1;
225   }
226 
227   /// Return the location number that matches Loc.
228   ///
229   /// For undef values we always return location number UndefLocNo without
230   /// inserting anything in locations. Since locations is a vector and the
231   /// location number is the position in the vector and UndefLocNo is ~0,
232   /// we would need a very big vector to put the value at the right position.
233   unsigned getLocationNo(const MachineOperand &LocMO) {
234     if (LocMO.isReg()) {
235       if (LocMO.getReg() == 0)
236         return UndefLocNo;
237       // For register locations we dont care about use/def and other flags.
238       for (unsigned i = 0, e = locations.size(); i != e; ++i)
239         if (locations[i].isReg() &&
240             locations[i].getReg() == LocMO.getReg() &&
241             locations[i].getSubReg() == LocMO.getSubReg())
242           return i;
243     } else
244       for (unsigned i = 0, e = locations.size(); i != e; ++i)
245         if (LocMO.isIdenticalTo(locations[i]))
246           return i;
247     locations.push_back(LocMO);
248     // We are storing a MachineOperand outside a MachineInstr.
249     locations.back().clearParent();
250     // Don't store def operands.
251     if (locations.back().isReg()) {
252       if (locations.back().isDef())
253         locations.back().setIsDead(false);
254       locations.back().setIsUse();
255     }
256     return locations.size() - 1;
257   }
258 
259   /// mapVirtRegs - Ensure that all virtual register locations are mapped.
260   void mapVirtRegs(LDVImpl *LDV);
261 
262   /// addDef - Add a definition point to this value.
263   void addDef(SlotIndex Idx, const MachineOperand &LocMO, bool IsIndirect) {
264     DbgValueLocation Loc(getLocationNo(LocMO), IsIndirect);
265     // Add a singular (Idx,Idx) -> Loc mapping.
266     LocMap::iterator I = locInts.find(Idx);
267     if (!I.valid() || I.start() != Idx)
268       I.insert(Idx, Idx.getNextSlot(), Loc);
269     else
270       // A later DBG_VALUE at the same SlotIndex overrides the old location.
271       I.setValue(Loc);
272   }
273 
274   /// extendDef - Extend the current definition as far as possible down.
275   /// Stop when meeting an existing def or when leaving the live
276   /// range of VNI.
277   /// End points where VNI is no longer live are added to Kills.
278   /// @param Idx   Starting point for the definition.
279   /// @param Loc   Location number to propagate.
280   /// @param LR    Restrict liveness to where LR has the value VNI. May be null.
281   /// @param VNI   When LR is not null, this is the value to restrict to.
282   /// @param Kills Append end points of VNI's live range to Kills.
283   /// @param LIS   Live intervals analysis.
284   void extendDef(SlotIndex Idx, DbgValueLocation Loc,
285                  LiveRange *LR, const VNInfo *VNI,
286                  SmallVectorImpl<SlotIndex> *Kills,
287                  LiveIntervals &LIS);
288 
289   /// addDefsFromCopies - The value in LI/LocNo may be copies to other
290   /// registers. Determine if any of the copies are available at the kill
291   /// points, and add defs if possible.
292   /// @param LI      Scan for copies of the value in LI->reg.
293   /// @param LocNo   Location number of LI->reg.
294   /// @param WasIndirect Indicates if the original use of LI->reg was indirect
295   /// @param Kills   Points where the range of LocNo could be extended.
296   /// @param NewDefs Append (Idx, LocNo) of inserted defs here.
297   void addDefsFromCopies(
298       LiveInterval *LI, unsigned LocNo, bool WasIndirect,
299       const SmallVectorImpl<SlotIndex> &Kills,
300       SmallVectorImpl<std::pair<SlotIndex, DbgValueLocation>> &NewDefs,
301       MachineRegisterInfo &MRI, LiveIntervals &LIS);
302 
303   /// computeIntervals - Compute the live intervals of all locations after
304   /// collecting all their def points.
305   void computeIntervals(MachineRegisterInfo &MRI, const TargetRegisterInfo &TRI,
306                         LiveIntervals &LIS, LexicalScopes &LS);
307 
308   /// splitRegister - Replace OldReg ranges with NewRegs ranges where NewRegs is
309   /// live. Returns true if any changes were made.
310   bool splitRegister(unsigned OldReg, ArrayRef<unsigned> NewRegs,
311                      LiveIntervals &LIS);
312 
313   /// rewriteLocations - Rewrite virtual register locations according to the
314   /// provided virtual register map. Record which locations were spilled.
315   void rewriteLocations(VirtRegMap &VRM, const TargetRegisterInfo &TRI,
316                         BitVector &SpilledLocations);
317 
318   /// emitDebugValues - Recreate DBG_VALUE instruction from data structures.
319   void emitDebugValues(VirtRegMap *VRM, LiveIntervals &LIS,
320                        const TargetInstrInfo &TII,
321                        const TargetRegisterInfo &TRI,
322                        const BitVector &SpilledLocations);
323 
324   /// getDebugLoc - Return DebugLoc of this UserValue.
325   DebugLoc getDebugLoc() { return dl;}
326 
327   void print(raw_ostream &, const TargetRegisterInfo *);
328 };
329 
330 /// LDVImpl - Implementation of the LiveDebugVariables pass.
331 class LDVImpl {
332   LiveDebugVariables &pass;
333   LocMap::Allocator allocator;
334   MachineFunction *MF = nullptr;
335   LiveIntervals *LIS;
336   const TargetRegisterInfo *TRI;
337 
338   /// Whether emitDebugValues is called.
339   bool EmitDone = false;
340 
341   /// Whether the machine function is modified during the pass.
342   bool ModifiedMF = false;
343 
344   /// userValues - All allocated UserValue instances.
345   SmallVector<std::unique_ptr<UserValue>, 8> userValues;
346 
347   /// Map virtual register to eq class leader.
348   using VRMap = DenseMap<unsigned, UserValue *>;
349   VRMap virtRegToEqClass;
350 
351   /// Map user variable to eq class leader.
352   using UVMap = DenseMap<const DILocalVariable *, UserValue *>;
353   UVMap userVarMap;
354 
355   /// getUserValue - Find or create a UserValue.
356   UserValue *getUserValue(const DILocalVariable *Var, const DIExpression *Expr,
357                           const DebugLoc &DL);
358 
359   /// lookupVirtReg - Find the EC leader for VirtReg or null.
360   UserValue *lookupVirtReg(unsigned VirtReg);
361 
362   /// handleDebugValue - Add DBG_VALUE instruction to our maps.
363   /// @param MI  DBG_VALUE instruction
364   /// @param Idx Last valid SLotIndex before instruction.
365   /// @return    True if the DBG_VALUE instruction should be deleted.
366   bool handleDebugValue(MachineInstr &MI, SlotIndex Idx);
367 
368   /// collectDebugValues - Collect and erase all DBG_VALUE instructions, adding
369   /// a UserValue def for each instruction.
370   /// @param mf MachineFunction to be scanned.
371   /// @return True if any debug values were found.
372   bool collectDebugValues(MachineFunction &mf);
373 
374   /// computeIntervals - Compute the live intervals of all user values after
375   /// collecting all their def points.
376   void computeIntervals();
377 
378 public:
379   LDVImpl(LiveDebugVariables *ps) : pass(*ps) {}
380 
381   bool runOnMachineFunction(MachineFunction &mf);
382 
383   /// clear - Release all memory.
384   void clear() {
385     MF = nullptr;
386     userValues.clear();
387     virtRegToEqClass.clear();
388     userVarMap.clear();
389     // Make sure we call emitDebugValues if the machine function was modified.
390     assert((!ModifiedMF || EmitDone) &&
391            "Dbg values are not emitted in LDV");
392     EmitDone = false;
393     ModifiedMF = false;
394   }
395 
396   /// mapVirtReg - Map virtual register to an equivalence class.
397   void mapVirtReg(unsigned VirtReg, UserValue *EC);
398 
399   /// splitRegister -  Replace all references to OldReg with NewRegs.
400   void splitRegister(unsigned OldReg, ArrayRef<unsigned> NewRegs);
401 
402   /// emitDebugValues - Recreate DBG_VALUE instruction from data structures.
403   void emitDebugValues(VirtRegMap *VRM);
404 
405   void print(raw_ostream&);
406 };
407 
408 } // end anonymous namespace
409 
410 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
411 static void printDebugLoc(const DebugLoc &DL, raw_ostream &CommentOS,
412                           const LLVMContext &Ctx) {
413   if (!DL)
414     return;
415 
416   auto *Scope = cast<DIScope>(DL.getScope());
417   // Omit the directory, because it's likely to be long and uninteresting.
418   CommentOS << Scope->getFilename();
419   CommentOS << ':' << DL.getLine();
420   if (DL.getCol() != 0)
421     CommentOS << ':' << DL.getCol();
422 
423   DebugLoc InlinedAtDL = DL.getInlinedAt();
424   if (!InlinedAtDL)
425     return;
426 
427   CommentOS << " @[ ";
428   printDebugLoc(InlinedAtDL, CommentOS, Ctx);
429   CommentOS << " ]";
430 }
431 
432 static void printExtendedName(raw_ostream &OS, const DILocalVariable *V,
433                               const DILocation *DL) {
434   const LLVMContext &Ctx = V->getContext();
435   StringRef Res = V->getName();
436   if (!Res.empty())
437     OS << Res << "," << V->getLine();
438   if (auto *InlinedAt = DL->getInlinedAt()) {
439     if (DebugLoc InlinedAtDL = InlinedAt) {
440       OS << " @[";
441       printDebugLoc(InlinedAtDL, OS, Ctx);
442       OS << "]";
443     }
444   }
445 }
446 
447 void UserValue::print(raw_ostream &OS, const TargetRegisterInfo *TRI) {
448   auto *DV = cast<DILocalVariable>(Variable);
449   OS << "!\"";
450   printExtendedName(OS, DV, dl);
451 
452   OS << "\"\t";
453   for (LocMap::const_iterator I = locInts.begin(); I.valid(); ++I) {
454     OS << " [" << I.start() << ';' << I.stop() << "):";
455     if (I.value().isUndef())
456       OS << "undef";
457     else {
458       OS << I.value().locNo();
459       if (I.value().wasIndirect())
460         OS << " ind";
461     }
462   }
463   for (unsigned i = 0, e = locations.size(); i != e; ++i) {
464     OS << " Loc" << i << '=';
465     locations[i].print(OS, TRI);
466   }
467   OS << '\n';
468 }
469 
470 void LDVImpl::print(raw_ostream &OS) {
471   OS << "********** DEBUG VARIABLES **********\n";
472   for (unsigned i = 0, e = userValues.size(); i != e; ++i)
473     userValues[i]->print(OS, TRI);
474 }
475 #endif
476 
477 void UserValue::mapVirtRegs(LDVImpl *LDV) {
478   for (unsigned i = 0, e = locations.size(); i != e; ++i)
479     if (locations[i].isReg() &&
480         TargetRegisterInfo::isVirtualRegister(locations[i].getReg()))
481       LDV->mapVirtReg(locations[i].getReg(), this);
482 }
483 
484 UserValue *LDVImpl::getUserValue(const DILocalVariable *Var,
485                                  const DIExpression *Expr, const DebugLoc &DL) {
486   UserValue *&Leader = userVarMap[Var];
487   if (Leader) {
488     UserValue *UV = Leader->getLeader();
489     Leader = UV;
490     for (; UV; UV = UV->getNext())
491       if (UV->match(Var, Expr, DL->getInlinedAt()))
492         return UV;
493   }
494 
495   userValues.push_back(
496       llvm::make_unique<UserValue>(Var, Expr, DL, allocator));
497   UserValue *UV = userValues.back().get();
498   Leader = UserValue::merge(Leader, UV);
499   return UV;
500 }
501 
502 void LDVImpl::mapVirtReg(unsigned VirtReg, UserValue *EC) {
503   assert(TargetRegisterInfo::isVirtualRegister(VirtReg) && "Only map VirtRegs");
504   UserValue *&Leader = virtRegToEqClass[VirtReg];
505   Leader = UserValue::merge(Leader, EC);
506 }
507 
508 UserValue *LDVImpl::lookupVirtReg(unsigned VirtReg) {
509   if (UserValue *UV = virtRegToEqClass.lookup(VirtReg))
510     return UV->getLeader();
511   return nullptr;
512 }
513 
514 bool LDVImpl::handleDebugValue(MachineInstr &MI, SlotIndex Idx) {
515   // DBG_VALUE loc, offset, variable
516   if (MI.getNumOperands() != 4 ||
517       !(MI.getOperand(1).isReg() || MI.getOperand(1).isImm()) ||
518       !MI.getOperand(2).isMetadata()) {
519     LLVM_DEBUG(dbgs() << "Can't handle " << MI);
520     return false;
521   }
522 
523   // Detect invalid DBG_VALUE instructions, with a debug-use of a virtual
524   // register that hasn't been defined yet. If we do not remove those here, then
525   // the re-insertion of the DBG_VALUE instruction after register allocation
526   // will be incorrect.
527   // TODO: If earlier passes are corrected to generate sane debug information
528   // (and if the machine verifier is improved to catch this), then these checks
529   // could be removed or replaced by asserts.
530   bool Discard = false;
531   if (MI.getOperand(0).isReg() &&
532       TargetRegisterInfo::isVirtualRegister(MI.getOperand(0).getReg())) {
533     const unsigned Reg = MI.getOperand(0).getReg();
534     if (!LIS->hasInterval(Reg)) {
535       // The DBG_VALUE is described by a virtual register that does not have a
536       // live interval. Discard the DBG_VALUE.
537       Discard = true;
538       LLVM_DEBUG(dbgs() << "Discarding debug info (no LIS interval): " << Idx
539                         << " " << MI);
540     } else {
541       // The DBG_VALUE is only valid if either Reg is live out from Idx, or Reg
542       // is defined dead at Idx (where Idx is the slot index for the instruction
543       // preceeding the DBG_VALUE).
544       const LiveInterval &LI = LIS->getInterval(Reg);
545       LiveQueryResult LRQ = LI.Query(Idx);
546       if (!LRQ.valueOutOrDead()) {
547         // We have found a DBG_VALUE with the value in a virtual register that
548         // is not live. Discard the DBG_VALUE.
549         Discard = true;
550         LLVM_DEBUG(dbgs() << "Discarding debug info (reg not live): " << Idx
551                           << " " << MI);
552       }
553     }
554   }
555 
556   // Get or create the UserValue for (variable,offset) here.
557   bool IsIndirect = MI.getOperand(1).isImm();
558   if (IsIndirect)
559     assert(MI.getOperand(1).getImm() == 0 && "DBG_VALUE with nonzero offset");
560   const DILocalVariable *Var = MI.getDebugVariable();
561   const DIExpression *Expr = MI.getDebugExpression();
562   UserValue *UV =
563       getUserValue(Var, Expr, MI.getDebugLoc());
564   if (!Discard)
565     UV->addDef(Idx, MI.getOperand(0), IsIndirect);
566   else {
567     MachineOperand MO = MachineOperand::CreateReg(0U, false);
568     MO.setIsDebug();
569     UV->addDef(Idx, MO, false);
570   }
571   return true;
572 }
573 
574 bool LDVImpl::collectDebugValues(MachineFunction &mf) {
575   bool Changed = false;
576   for (MachineFunction::iterator MFI = mf.begin(), MFE = mf.end(); MFI != MFE;
577        ++MFI) {
578     MachineBasicBlock *MBB = &*MFI;
579     for (MachineBasicBlock::iterator MBBI = MBB->begin(), MBBE = MBB->end();
580          MBBI != MBBE;) {
581       // Use the first debug instruction in the sequence to get a SlotIndex
582       // for following consecutive debug instructions.
583       if (!MBBI->isDebugInstr()) {
584         ++MBBI;
585         continue;
586       }
587       // Debug instructions has no slot index. Use the previous
588       // non-debug instruction's SlotIndex as its SlotIndex.
589       SlotIndex Idx =
590           MBBI == MBB->begin()
591               ? LIS->getMBBStartIdx(MBB)
592               : LIS->getInstructionIndex(*std::prev(MBBI)).getRegSlot();
593       // Handle consecutive debug instructions with the same slot index.
594       do {
595         // Only handle DBG_VALUE in handleDebugValue(). Skip all other
596         // kinds of debug instructions.
597         if (MBBI->isDebugValue() && handleDebugValue(*MBBI, Idx)) {
598           MBBI = MBB->erase(MBBI);
599           Changed = true;
600         } else
601           ++MBBI;
602       } while (MBBI != MBBE && MBBI->isDebugInstr());
603     }
604   }
605   return Changed;
606 }
607 
608 /// We only propagate DBG_VALUES locally here. LiveDebugValues performs a
609 /// data-flow analysis to propagate them beyond basic block boundaries.
610 void UserValue::extendDef(SlotIndex Idx, DbgValueLocation Loc, LiveRange *LR,
611                           const VNInfo *VNI, SmallVectorImpl<SlotIndex> *Kills,
612                           LiveIntervals &LIS) {
613   SlotIndex Start = Idx;
614   MachineBasicBlock *MBB = LIS.getMBBFromIndex(Start);
615   SlotIndex Stop = LIS.getMBBEndIdx(MBB);
616   LocMap::iterator I = locInts.find(Start);
617 
618   // Limit to VNI's live range.
619   bool ToEnd = true;
620   if (LR && VNI) {
621     LiveInterval::Segment *Segment = LR->getSegmentContaining(Start);
622     if (!Segment || Segment->valno != VNI) {
623       if (Kills)
624         Kills->push_back(Start);
625       return;
626     }
627     if (Segment->end < Stop) {
628       Stop = Segment->end;
629       ToEnd = false;
630     }
631   }
632 
633   // There could already be a short def at Start.
634   if (I.valid() && I.start() <= Start) {
635     // Stop when meeting a different location or an already extended interval.
636     Start = Start.getNextSlot();
637     if (I.value() != Loc || I.stop() != Start)
638       return;
639     // This is a one-slot placeholder. Just skip it.
640     ++I;
641   }
642 
643   // Limited by the next def.
644   if (I.valid() && I.start() < Stop) {
645     Stop = I.start();
646     ToEnd = false;
647   }
648   // Limited by VNI's live range.
649   else if (!ToEnd && Kills)
650     Kills->push_back(Stop);
651 
652   if (Start < Stop)
653     I.insert(Start, Stop, Loc);
654 }
655 
656 void UserValue::addDefsFromCopies(
657     LiveInterval *LI, unsigned LocNo, bool WasIndirect,
658     const SmallVectorImpl<SlotIndex> &Kills,
659     SmallVectorImpl<std::pair<SlotIndex, DbgValueLocation>> &NewDefs,
660     MachineRegisterInfo &MRI, LiveIntervals &LIS) {
661   if (Kills.empty())
662     return;
663   // Don't track copies from physregs, there are too many uses.
664   if (!TargetRegisterInfo::isVirtualRegister(LI->reg))
665     return;
666 
667   // Collect all the (vreg, valno) pairs that are copies of LI.
668   SmallVector<std::pair<LiveInterval*, const VNInfo*>, 8> CopyValues;
669   for (MachineOperand &MO : MRI.use_nodbg_operands(LI->reg)) {
670     MachineInstr *MI = MO.getParent();
671     // Copies of the full value.
672     if (MO.getSubReg() || !MI->isCopy())
673       continue;
674     unsigned DstReg = MI->getOperand(0).getReg();
675 
676     // Don't follow copies to physregs. These are usually setting up call
677     // arguments, and the argument registers are always call clobbered. We are
678     // better off in the source register which could be a callee-saved register,
679     // or it could be spilled.
680     if (!TargetRegisterInfo::isVirtualRegister(DstReg))
681       continue;
682 
683     // Is LocNo extended to reach this copy? If not, another def may be blocking
684     // it, or we are looking at a wrong value of LI.
685     SlotIndex Idx = LIS.getInstructionIndex(*MI);
686     LocMap::iterator I = locInts.find(Idx.getRegSlot(true));
687     if (!I.valid() || I.value().locNo() != LocNo)
688       continue;
689 
690     if (!LIS.hasInterval(DstReg))
691       continue;
692     LiveInterval *DstLI = &LIS.getInterval(DstReg);
693     const VNInfo *DstVNI = DstLI->getVNInfoAt(Idx.getRegSlot());
694     assert(DstVNI && DstVNI->def == Idx.getRegSlot() && "Bad copy value");
695     CopyValues.push_back(std::make_pair(DstLI, DstVNI));
696   }
697 
698   if (CopyValues.empty())
699     return;
700 
701   LLVM_DEBUG(dbgs() << "Got " << CopyValues.size() << " copies of " << *LI
702                     << '\n');
703 
704   // Try to add defs of the copied values for each kill point.
705   for (unsigned i = 0, e = Kills.size(); i != e; ++i) {
706     SlotIndex Idx = Kills[i];
707     for (unsigned j = 0, e = CopyValues.size(); j != e; ++j) {
708       LiveInterval *DstLI = CopyValues[j].first;
709       const VNInfo *DstVNI = CopyValues[j].second;
710       if (DstLI->getVNInfoAt(Idx) != DstVNI)
711         continue;
712       // Check that there isn't already a def at Idx
713       LocMap::iterator I = locInts.find(Idx);
714       if (I.valid() && I.start() <= Idx)
715         continue;
716       LLVM_DEBUG(dbgs() << "Kill at " << Idx << " covered by valno #"
717                         << DstVNI->id << " in " << *DstLI << '\n');
718       MachineInstr *CopyMI = LIS.getInstructionFromIndex(DstVNI->def);
719       assert(CopyMI && CopyMI->isCopy() && "Bad copy value");
720       unsigned LocNo = getLocationNo(CopyMI->getOperand(0));
721       DbgValueLocation NewLoc(LocNo, WasIndirect);
722       I.insert(Idx, Idx.getNextSlot(), NewLoc);
723       NewDefs.push_back(std::make_pair(Idx, NewLoc));
724       break;
725     }
726   }
727 }
728 
729 void UserValue::computeIntervals(MachineRegisterInfo &MRI,
730                                  const TargetRegisterInfo &TRI,
731                                  LiveIntervals &LIS, LexicalScopes &LS) {
732   SmallVector<std::pair<SlotIndex, DbgValueLocation>, 16> Defs;
733 
734   // Collect all defs to be extended (Skipping undefs).
735   for (LocMap::const_iterator I = locInts.begin(); I.valid(); ++I)
736     if (!I.value().isUndef())
737       Defs.push_back(std::make_pair(I.start(), I.value()));
738 
739   // Extend all defs, and possibly add new ones along the way.
740   for (unsigned i = 0; i != Defs.size(); ++i) {
741     SlotIndex Idx = Defs[i].first;
742     DbgValueLocation Loc = Defs[i].second;
743     const MachineOperand &LocMO = locations[Loc.locNo()];
744 
745     if (!LocMO.isReg()) {
746       extendDef(Idx, Loc, nullptr, nullptr, nullptr, LIS);
747       continue;
748     }
749 
750     // Register locations are constrained to where the register value is live.
751     if (TargetRegisterInfo::isVirtualRegister(LocMO.getReg())) {
752       LiveInterval *LI = nullptr;
753       const VNInfo *VNI = nullptr;
754       if (LIS.hasInterval(LocMO.getReg())) {
755         LI = &LIS.getInterval(LocMO.getReg());
756         VNI = LI->getVNInfoAt(Idx);
757       }
758       SmallVector<SlotIndex, 16> Kills;
759       extendDef(Idx, Loc, LI, VNI, &Kills, LIS);
760       // FIXME: Handle sub-registers in addDefsFromCopies. The problem is that
761       // if the original location for example is %vreg0:sub_hi, and we find a
762       // full register copy in addDefsFromCopies (at the moment it only handles
763       // full register copies), then we must add the sub1 sub-register index to
764       // the new location. However, that is only possible if the new virtual
765       // register is of the same regclass (or if there is an equivalent
766       // sub-register in that regclass). For now, simply skip handling copies if
767       // a sub-register is involved.
768       if (LI && !LocMO.getSubReg())
769         addDefsFromCopies(LI, Loc.locNo(), Loc.wasIndirect(), Kills, Defs, MRI,
770                           LIS);
771       continue;
772     }
773 
774     // For physregs, we only mark the start slot idx. DwarfDebug will see it
775     // as if the DBG_VALUE is valid up until the end of the basic block, or
776     // the next def of the physical register. So we do not need to extend the
777     // range. It might actually happen that the DBG_VALUE is the last use of
778     // the physical register (e.g. if this is an unused input argument to a
779     // function).
780   }
781 
782   // The computed intervals may extend beyond the range of the debug
783   // location's lexical scope. In this case, splitting of an interval
784   // can result in an interval outside of the scope being created,
785   // causing extra unnecessary DBG_VALUEs to be emitted. To prevent
786   // this, trim the intervals to the lexical scope.
787 
788   LexicalScope *Scope = LS.findLexicalScope(dl);
789   if (!Scope)
790     return;
791 
792   SlotIndex PrevEnd;
793   LocMap::iterator I = locInts.begin();
794 
795   // Iterate over the lexical scope ranges. Each time round the loop
796   // we check the intervals for overlap with the end of the previous
797   // range and the start of the next. The first range is handled as
798   // a special case where there is no PrevEnd.
799   for (const InsnRange &Range : Scope->getRanges()) {
800     SlotIndex RStart = LIS.getInstructionIndex(*Range.first);
801     SlotIndex REnd = LIS.getInstructionIndex(*Range.second);
802 
803     // At the start of each iteration I has been advanced so that
804     // I.stop() >= PrevEnd. Check for overlap.
805     if (PrevEnd && I.start() < PrevEnd) {
806       SlotIndex IStop = I.stop();
807       DbgValueLocation Loc = I.value();
808 
809       // Stop overlaps previous end - trim the end of the interval to
810       // the scope range.
811       I.setStopUnchecked(PrevEnd);
812       ++I;
813 
814       // If the interval also overlaps the start of the "next" (i.e.
815       // current) range create a new interval for the remainder (which
816       // may be further trimmed).
817       if (RStart < IStop)
818         I.insert(RStart, IStop, Loc);
819     }
820 
821     // Advance I so that I.stop() >= RStart, and check for overlap.
822     I.advanceTo(RStart);
823     if (!I.valid())
824       return;
825 
826     if (I.start() < RStart) {
827       // Interval start overlaps range - trim to the scope range.
828       I.setStartUnchecked(RStart);
829       // Remember that this interval was trimmed.
830       trimmedDefs.insert(RStart);
831     }
832 
833     // The end of a lexical scope range is the last instruction in the
834     // range. To convert to an interval we need the index of the
835     // instruction after it.
836     REnd = REnd.getNextIndex();
837 
838     // Advance I to first interval outside current range.
839     I.advanceTo(REnd);
840     if (!I.valid())
841       return;
842 
843     PrevEnd = REnd;
844   }
845 
846   // Check for overlap with end of final range.
847   if (PrevEnd && I.start() < PrevEnd)
848     I.setStopUnchecked(PrevEnd);
849 }
850 
851 void LDVImpl::computeIntervals() {
852   LexicalScopes LS;
853   LS.initialize(*MF);
854 
855   for (unsigned i = 0, e = userValues.size(); i != e; ++i) {
856     userValues[i]->computeIntervals(MF->getRegInfo(), *TRI, *LIS, LS);
857     userValues[i]->mapVirtRegs(this);
858   }
859 }
860 
861 bool LDVImpl::runOnMachineFunction(MachineFunction &mf) {
862   clear();
863   MF = &mf;
864   LIS = &pass.getAnalysis<LiveIntervals>();
865   TRI = mf.getSubtarget().getRegisterInfo();
866   LLVM_DEBUG(dbgs() << "********** COMPUTING LIVE DEBUG VARIABLES: "
867                     << mf.getName() << " **********\n");
868 
869   bool Changed = collectDebugValues(mf);
870   computeIntervals();
871   LLVM_DEBUG(print(dbgs()));
872   ModifiedMF = Changed;
873   return Changed;
874 }
875 
876 static void removeDebugValues(MachineFunction &mf) {
877   for (MachineBasicBlock &MBB : mf) {
878     for (auto MBBI = MBB.begin(), MBBE = MBB.end(); MBBI != MBBE; ) {
879       if (!MBBI->isDebugValue()) {
880         ++MBBI;
881         continue;
882       }
883       MBBI = MBB.erase(MBBI);
884     }
885   }
886 }
887 
888 bool LiveDebugVariables::runOnMachineFunction(MachineFunction &mf) {
889   if (!EnableLDV)
890     return false;
891   if (!mf.getFunction().getSubprogram()) {
892     removeDebugValues(mf);
893     return false;
894   }
895   if (!pImpl)
896     pImpl = new LDVImpl(this);
897   return static_cast<LDVImpl*>(pImpl)->runOnMachineFunction(mf);
898 }
899 
900 void LiveDebugVariables::releaseMemory() {
901   if (pImpl)
902     static_cast<LDVImpl*>(pImpl)->clear();
903 }
904 
905 LiveDebugVariables::~LiveDebugVariables() {
906   if (pImpl)
907     delete static_cast<LDVImpl*>(pImpl);
908 }
909 
910 //===----------------------------------------------------------------------===//
911 //                           Live Range Splitting
912 //===----------------------------------------------------------------------===//
913 
914 bool
915 UserValue::splitLocation(unsigned OldLocNo, ArrayRef<unsigned> NewRegs,
916                          LiveIntervals& LIS) {
917   LLVM_DEBUG({
918     dbgs() << "Splitting Loc" << OldLocNo << '\t';
919     print(dbgs(), nullptr);
920   });
921   bool DidChange = false;
922   LocMap::iterator LocMapI;
923   LocMapI.setMap(locInts);
924   for (unsigned i = 0; i != NewRegs.size(); ++i) {
925     LiveInterval *LI = &LIS.getInterval(NewRegs[i]);
926     if (LI->empty())
927       continue;
928 
929     // Don't allocate the new LocNo until it is needed.
930     unsigned NewLocNo = UndefLocNo;
931 
932     // Iterate over the overlaps between locInts and LI.
933     LocMapI.find(LI->beginIndex());
934     if (!LocMapI.valid())
935       continue;
936     LiveInterval::iterator LII = LI->advanceTo(LI->begin(), LocMapI.start());
937     LiveInterval::iterator LIE = LI->end();
938     while (LocMapI.valid() && LII != LIE) {
939       // At this point, we know that LocMapI.stop() > LII->start.
940       LII = LI->advanceTo(LII, LocMapI.start());
941       if (LII == LIE)
942         break;
943 
944       // Now LII->end > LocMapI.start(). Do we have an overlap?
945       if (LocMapI.value().locNo() == OldLocNo && LII->start < LocMapI.stop()) {
946         // Overlapping correct location. Allocate NewLocNo now.
947         if (NewLocNo == UndefLocNo) {
948           MachineOperand MO = MachineOperand::CreateReg(LI->reg, false);
949           MO.setSubReg(locations[OldLocNo].getSubReg());
950           NewLocNo = getLocationNo(MO);
951           DidChange = true;
952         }
953 
954         SlotIndex LStart = LocMapI.start();
955         SlotIndex LStop  = LocMapI.stop();
956         DbgValueLocation OldLoc = LocMapI.value();
957 
958         // Trim LocMapI down to the LII overlap.
959         if (LStart < LII->start)
960           LocMapI.setStartUnchecked(LII->start);
961         if (LStop > LII->end)
962           LocMapI.setStopUnchecked(LII->end);
963 
964         // Change the value in the overlap. This may trigger coalescing.
965         LocMapI.setValue(OldLoc.changeLocNo(NewLocNo));
966 
967         // Re-insert any removed OldLocNo ranges.
968         if (LStart < LocMapI.start()) {
969           LocMapI.insert(LStart, LocMapI.start(), OldLoc);
970           ++LocMapI;
971           assert(LocMapI.valid() && "Unexpected coalescing");
972         }
973         if (LStop > LocMapI.stop()) {
974           ++LocMapI;
975           LocMapI.insert(LII->end, LStop, OldLoc);
976           --LocMapI;
977         }
978       }
979 
980       // Advance to the next overlap.
981       if (LII->end < LocMapI.stop()) {
982         if (++LII == LIE)
983           break;
984         LocMapI.advanceTo(LII->start);
985       } else {
986         ++LocMapI;
987         if (!LocMapI.valid())
988           break;
989         LII = LI->advanceTo(LII, LocMapI.start());
990       }
991     }
992   }
993 
994   // Finally, remove any remaining OldLocNo intervals and OldLocNo itself.
995   locations.erase(locations.begin() + OldLocNo);
996   LocMapI.goToBegin();
997   while (LocMapI.valid()) {
998     DbgValueLocation v = LocMapI.value();
999     if (v.locNo() == OldLocNo) {
1000       LLVM_DEBUG(dbgs() << "Erasing [" << LocMapI.start() << ';'
1001                         << LocMapI.stop() << ")\n");
1002       LocMapI.erase();
1003     } else {
1004       // Undef values always have location number UndefLocNo, so don't change
1005       // locNo in that case. See getLocationNo().
1006       if (!v.isUndef() && v.locNo() > OldLocNo)
1007         LocMapI.setValueUnchecked(v.changeLocNo(v.locNo() - 1));
1008       ++LocMapI;
1009     }
1010   }
1011 
1012   LLVM_DEBUG({
1013     dbgs() << "Split result: \t";
1014     print(dbgs(), nullptr);
1015   });
1016   return DidChange;
1017 }
1018 
1019 bool
1020 UserValue::splitRegister(unsigned OldReg, ArrayRef<unsigned> NewRegs,
1021                          LiveIntervals &LIS) {
1022   bool DidChange = false;
1023   // Split locations referring to OldReg. Iterate backwards so splitLocation can
1024   // safely erase unused locations.
1025   for (unsigned i = locations.size(); i ; --i) {
1026     unsigned LocNo = i-1;
1027     const MachineOperand *Loc = &locations[LocNo];
1028     if (!Loc->isReg() || Loc->getReg() != OldReg)
1029       continue;
1030     DidChange |= splitLocation(LocNo, NewRegs, LIS);
1031   }
1032   return DidChange;
1033 }
1034 
1035 void LDVImpl::splitRegister(unsigned OldReg, ArrayRef<unsigned> NewRegs) {
1036   bool DidChange = false;
1037   for (UserValue *UV = lookupVirtReg(OldReg); UV; UV = UV->getNext())
1038     DidChange |= UV->splitRegister(OldReg, NewRegs, *LIS);
1039 
1040   if (!DidChange)
1041     return;
1042 
1043   // Map all of the new virtual registers.
1044   UserValue *UV = lookupVirtReg(OldReg);
1045   for (unsigned i = 0; i != NewRegs.size(); ++i)
1046     mapVirtReg(NewRegs[i], UV);
1047 }
1048 
1049 void LiveDebugVariables::
1050 splitRegister(unsigned OldReg, ArrayRef<unsigned> NewRegs, LiveIntervals &LIS) {
1051   if (pImpl)
1052     static_cast<LDVImpl*>(pImpl)->splitRegister(OldReg, NewRegs);
1053 }
1054 
1055 void UserValue::rewriteLocations(VirtRegMap &VRM, const TargetRegisterInfo &TRI,
1056                                  BitVector &SpilledLocations) {
1057   // Build a set of new locations with new numbers so we can coalesce our
1058   // IntervalMap if two vreg intervals collapse to the same physical location.
1059   // Use MapVector instead of SetVector because MapVector::insert returns the
1060   // position of the previously or newly inserted element. The boolean value
1061   // tracks if the location was produced by a spill.
1062   // FIXME: This will be problematic if we ever support direct and indirect
1063   // frame index locations, i.e. expressing both variables in memory and
1064   // 'int x, *px = &x'. The "spilled" bit must become part of the location.
1065   MapVector<MachineOperand, bool> NewLocations;
1066   SmallVector<unsigned, 4> LocNoMap(locations.size());
1067   for (unsigned I = 0, E = locations.size(); I != E; ++I) {
1068     bool Spilled = false;
1069     MachineOperand Loc = locations[I];
1070     // Only virtual registers are rewritten.
1071     if (Loc.isReg() && Loc.getReg() &&
1072         TargetRegisterInfo::isVirtualRegister(Loc.getReg())) {
1073       unsigned VirtReg = Loc.getReg();
1074       if (VRM.isAssignedReg(VirtReg) &&
1075           TargetRegisterInfo::isPhysicalRegister(VRM.getPhys(VirtReg))) {
1076         // This can create a %noreg operand in rare cases when the sub-register
1077         // index is no longer available. That means the user value is in a
1078         // non-existent sub-register, and %noreg is exactly what we want.
1079         Loc.substPhysReg(VRM.getPhys(VirtReg), TRI);
1080       } else if (VRM.getStackSlot(VirtReg) != VirtRegMap::NO_STACK_SLOT) {
1081         // FIXME: Translate SubIdx to a stackslot offset.
1082         Loc = MachineOperand::CreateFI(VRM.getStackSlot(VirtReg));
1083         Spilled = true;
1084       } else {
1085         Loc.setReg(0);
1086         Loc.setSubReg(0);
1087       }
1088     }
1089 
1090     // Insert this location if it doesn't already exist and record a mapping
1091     // from the old number to the new number.
1092     auto InsertResult = NewLocations.insert({Loc, Spilled});
1093     unsigned NewLocNo = std::distance(NewLocations.begin(), InsertResult.first);
1094     LocNoMap[I] = NewLocNo;
1095   }
1096 
1097   // Rewrite the locations and record which ones were spill slots.
1098   locations.clear();
1099   SpilledLocations.clear();
1100   SpilledLocations.resize(NewLocations.size());
1101   for (auto &Pair : NewLocations) {
1102     locations.push_back(Pair.first);
1103     if (Pair.second) {
1104       unsigned NewLocNo = std::distance(&*NewLocations.begin(), &Pair);
1105       SpilledLocations.set(NewLocNo);
1106     }
1107   }
1108 
1109   // Update the interval map, but only coalesce left, since intervals to the
1110   // right use the old location numbers. This should merge two contiguous
1111   // DBG_VALUE intervals with different vregs that were allocated to the same
1112   // physical register.
1113   for (LocMap::iterator I = locInts.begin(); I.valid(); ++I) {
1114     DbgValueLocation Loc = I.value();
1115     // Undef values don't exist in locations (and thus not in LocNoMap either)
1116     // so skip over them. See getLocationNo().
1117     if (Loc.isUndef())
1118       continue;
1119     unsigned NewLocNo = LocNoMap[Loc.locNo()];
1120     I.setValueUnchecked(Loc.changeLocNo(NewLocNo));
1121     I.setStart(I.start());
1122   }
1123 }
1124 
1125 /// Find an iterator for inserting a DBG_VALUE instruction.
1126 static MachineBasicBlock::iterator
1127 findInsertLocation(MachineBasicBlock *MBB, SlotIndex Idx,
1128                    LiveIntervals &LIS) {
1129   SlotIndex Start = LIS.getMBBStartIdx(MBB);
1130   Idx = Idx.getBaseIndex();
1131 
1132   // Try to find an insert location by going backwards from Idx.
1133   MachineInstr *MI;
1134   while (!(MI = LIS.getInstructionFromIndex(Idx))) {
1135     // We've reached the beginning of MBB.
1136     if (Idx == Start) {
1137       MachineBasicBlock::iterator I = MBB->SkipPHIsLabelsAndDebug(MBB->begin());
1138       return I;
1139     }
1140     Idx = Idx.getPrevIndex();
1141   }
1142 
1143   // Don't insert anything after the first terminator, though.
1144   return MI->isTerminator() ? MBB->getFirstTerminator() :
1145                               std::next(MachineBasicBlock::iterator(MI));
1146 }
1147 
1148 /// Find an iterator for inserting the next DBG_VALUE instruction
1149 /// (or end if no more insert locations found).
1150 static MachineBasicBlock::iterator
1151 findNextInsertLocation(MachineBasicBlock *MBB,
1152                        MachineBasicBlock::iterator I,
1153                        SlotIndex StopIdx, MachineOperand &LocMO,
1154                        LiveIntervals &LIS,
1155                        const TargetRegisterInfo &TRI) {
1156   if (!LocMO.isReg())
1157     return MBB->instr_end();
1158   unsigned Reg = LocMO.getReg();
1159 
1160   // Find the next instruction in the MBB that define the register Reg.
1161   while (I != MBB->end() && !I->isTerminator()) {
1162     if (!LIS.isNotInMIMap(*I) &&
1163         SlotIndex::isEarlierEqualInstr(StopIdx, LIS.getInstructionIndex(*I)))
1164       break;
1165     if (I->definesRegister(Reg, &TRI))
1166       // The insert location is directly after the instruction/bundle.
1167       return std::next(I);
1168     ++I;
1169   }
1170   return MBB->end();
1171 }
1172 
1173 void UserValue::insertDebugValue(MachineBasicBlock *MBB, SlotIndex StartIdx,
1174                                  SlotIndex StopIdx,
1175                                  DbgValueLocation Loc, bool Spilled,
1176                                  LiveIntervals &LIS,
1177                                  const TargetInstrInfo &TII,
1178                                  const TargetRegisterInfo &TRI) {
1179   SlotIndex MBBEndIdx = LIS.getMBBEndIdx(&*MBB);
1180   // Only search within the current MBB.
1181   StopIdx = (MBBEndIdx < StopIdx) ? MBBEndIdx : StopIdx;
1182   MachineBasicBlock::iterator I = findInsertLocation(MBB, StartIdx, LIS);
1183   // Undef values don't exist in locations so create new "noreg" register MOs
1184   // for them. See getLocationNo().
1185   MachineOperand MO = !Loc.isUndef() ?
1186     locations[Loc.locNo()] :
1187     MachineOperand::CreateReg(/* Reg */ 0, /* isDef */ false, /* isImp */ false,
1188                               /* isKill */ false, /* isDead */ false,
1189                               /* isUndef */ false, /* isEarlyClobber */ false,
1190                               /* SubReg */ 0, /* isDebug */ true);
1191 
1192   ++NumInsertedDebugValues;
1193 
1194   assert(cast<DILocalVariable>(Variable)
1195              ->isValidLocationForIntrinsic(getDebugLoc()) &&
1196          "Expected inlined-at fields to agree");
1197 
1198   // If the location was spilled, the new DBG_VALUE will be indirect. If the
1199   // original DBG_VALUE was indirect, we need to add DW_OP_deref to indicate
1200   // that the original virtual register was a pointer.
1201   const DIExpression *Expr = Expression;
1202   bool IsIndirect = Loc.wasIndirect();
1203   if (Spilled) {
1204     if (IsIndirect)
1205       Expr = DIExpression::prepend(Expr, DIExpression::WithDeref);
1206     IsIndirect = true;
1207   }
1208 
1209   assert((!Spilled || MO.isFI()) && "a spilled location must be a frame index");
1210 
1211   do {
1212     BuildMI(*MBB, I, getDebugLoc(), TII.get(TargetOpcode::DBG_VALUE),
1213             IsIndirect, MO, Variable, Expr);
1214 
1215     // Continue and insert DBG_VALUES after every redefinition of register
1216     // associated with the debug value within the range
1217     I = findNextInsertLocation(MBB, I, StopIdx, MO, LIS, TRI);
1218   } while (I != MBB->end());
1219 }
1220 
1221 void UserValue::emitDebugValues(VirtRegMap *VRM, LiveIntervals &LIS,
1222                                 const TargetInstrInfo &TII,
1223                                 const TargetRegisterInfo &TRI,
1224                                 const BitVector &SpilledLocations) {
1225   MachineFunction::iterator MFEnd = VRM->getMachineFunction().end();
1226 
1227   for (LocMap::const_iterator I = locInts.begin(); I.valid();) {
1228     SlotIndex Start = I.start();
1229     SlotIndex Stop = I.stop();
1230     DbgValueLocation Loc = I.value();
1231     bool Spilled = !Loc.isUndef() ? SpilledLocations.test(Loc.locNo()) : false;
1232 
1233     // If the interval start was trimmed to the lexical scope insert the
1234     // DBG_VALUE at the previous index (otherwise it appears after the
1235     // first instruction in the range).
1236     if (trimmedDefs.count(Start))
1237       Start = Start.getPrevIndex();
1238 
1239     LLVM_DEBUG(dbgs() << "\t[" << Start << ';' << Stop << "):" << Loc.locNo());
1240     MachineFunction::iterator MBB = LIS.getMBBFromIndex(Start)->getIterator();
1241     SlotIndex MBBEnd = LIS.getMBBEndIdx(&*MBB);
1242 
1243     LLVM_DEBUG(dbgs() << ' ' << printMBBReference(*MBB) << '-' << MBBEnd);
1244     insertDebugValue(&*MBB, Start, Stop, Loc, Spilled, LIS, TII, TRI);
1245     // This interval may span multiple basic blocks.
1246     // Insert a DBG_VALUE into each one.
1247     while (Stop > MBBEnd) {
1248       // Move to the next block.
1249       Start = MBBEnd;
1250       if (++MBB == MFEnd)
1251         break;
1252       MBBEnd = LIS.getMBBEndIdx(&*MBB);
1253       LLVM_DEBUG(dbgs() << ' ' << printMBBReference(*MBB) << '-' << MBBEnd);
1254       insertDebugValue(&*MBB, Start, Stop, Loc, Spilled, LIS, TII, TRI);
1255     }
1256     LLVM_DEBUG(dbgs() << '\n');
1257     if (MBB == MFEnd)
1258       break;
1259 
1260     ++I;
1261   }
1262 }
1263 
1264 void LDVImpl::emitDebugValues(VirtRegMap *VRM) {
1265   LLVM_DEBUG(dbgs() << "********** EMITTING LIVE DEBUG VARIABLES **********\n");
1266   if (!MF)
1267     return;
1268   const TargetInstrInfo *TII = MF->getSubtarget().getInstrInfo();
1269   BitVector SpilledLocations;
1270   for (unsigned i = 0, e = userValues.size(); i != e; ++i) {
1271     LLVM_DEBUG(userValues[i]->print(dbgs(), TRI));
1272     userValues[i]->rewriteLocations(*VRM, *TRI, SpilledLocations);
1273     userValues[i]->emitDebugValues(VRM, *LIS, *TII, *TRI, SpilledLocations);
1274   }
1275   EmitDone = true;
1276 }
1277 
1278 void LiveDebugVariables::emitDebugValues(VirtRegMap *VRM) {
1279   if (pImpl)
1280     static_cast<LDVImpl*>(pImpl)->emitDebugValues(VRM);
1281 }
1282 
1283 bool LiveDebugVariables::doInitialization(Module &M) {
1284   return Pass::doInitialization(M);
1285 }
1286 
1287 #if !defined(NDEBUG) || defined(LLVM_ENABLE_DUMP)
1288 LLVM_DUMP_METHOD void LiveDebugVariables::dump() const {
1289   if (pImpl)
1290     static_cast<LDVImpl*>(pImpl)->print(dbgs());
1291 }
1292 #endif
1293