DebugInfo: more support for mutating DICompositeType to reduce magic number usage...
[oota-llvm.git] / 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/DebugInfo.h"
16 #include "llvm/ADT/STLExtras.h"
17 #include "llvm/ADT/SmallPtrSet.h"
18 #include "llvm/ADT/SmallString.h"
19 #include "llvm/Analysis/ValueTracking.h"
20 #include "llvm/IR/Constants.h"
21 #include "llvm/IR/DerivedTypes.h"
22 #include "llvm/IR/Instructions.h"
23 #include "llvm/IR/IntrinsicInst.h"
24 #include "llvm/IR/Intrinsics.h"
25 #include "llvm/IR/Module.h"
26 #include "llvm/Support/Debug.h"
27 #include "llvm/Support/Dwarf.h"
28 #include "llvm/Support/ValueHandle.h"
29 #include "llvm/Support/raw_ostream.h"
30 using namespace llvm;
31 using namespace llvm::dwarf;
32
33 //===----------------------------------------------------------------------===//
34 // DIDescriptor
35 //===----------------------------------------------------------------------===//
36
37 DIDescriptor::DIDescriptor(const DIFile F) : DbgNode(F.DbgNode) {
38 }
39
40 DIDescriptor::DIDescriptor(const DISubprogram F) : DbgNode(F.DbgNode) {
41 }
42
43 DIDescriptor::DIDescriptor(const DILexicalBlockFile F) : DbgNode(F.DbgNode) {
44 }
45
46 DIDescriptor::DIDescriptor(const DILexicalBlock F) : DbgNode(F.DbgNode) {
47 }
48
49 DIDescriptor::DIDescriptor(const DIVariable F) : DbgNode(F.DbgNode) {
50 }
51
52 DIDescriptor::DIDescriptor(const DIType F) : DbgNode(F.DbgNode) {
53 }
54
55 bool DIDescriptor::Verify() const {
56   return DbgNode &&
57          (DIDerivedType(DbgNode).Verify() ||
58           DICompositeType(DbgNode).Verify() || DIBasicType(DbgNode).Verify() ||
59           DIVariable(DbgNode).Verify() || DISubprogram(DbgNode).Verify() ||
60           DIGlobalVariable(DbgNode).Verify() || DIFile(DbgNode).Verify() ||
61           DICompileUnit(DbgNode).Verify() || DINameSpace(DbgNode).Verify() ||
62           DILexicalBlock(DbgNode).Verify() ||
63           DILexicalBlockFile(DbgNode).Verify() ||
64           DISubrange(DbgNode).Verify() || DIEnumerator(DbgNode).Verify() ||
65           DIObjCProperty(DbgNode).Verify() ||
66           DITemplateTypeParameter(DbgNode).Verify() ||
67           DITemplateValueParameter(DbgNode).Verify());
68 }
69
70 static Value *getField(const MDNode *DbgNode, unsigned Elt) {
71   if (DbgNode == 0 || Elt >= DbgNode->getNumOperands())
72     return 0;
73   return DbgNode->getOperand(Elt);
74 }
75
76 static const MDNode *getNodeField(const MDNode *DbgNode, unsigned Elt) {
77   if (const MDNode *R = dyn_cast_or_null<MDNode>(getField(DbgNode, Elt)))
78     return R;
79   return 0;
80 }
81
82 static StringRef getStringField(const MDNode *DbgNode, unsigned Elt) {
83   if (MDString *MDS = dyn_cast_or_null<MDString>(getField(DbgNode, Elt)))
84     return MDS->getString();
85   return StringRef();
86 }
87
88 StringRef DIDescriptor::getStringField(unsigned Elt) const {
89   return ::getStringField(DbgNode, Elt);
90 }
91
92 uint64_t DIDescriptor::getUInt64Field(unsigned Elt) const {
93   if (DbgNode == 0)
94     return 0;
95
96   if (Elt < DbgNode->getNumOperands())
97     if (ConstantInt *CI
98         = dyn_cast_or_null<ConstantInt>(DbgNode->getOperand(Elt)))
99       return CI->getZExtValue();
100
101   return 0;
102 }
103
104 int64_t DIDescriptor::getInt64Field(unsigned Elt) const {
105   if (DbgNode == 0)
106     return 0;
107
108   if (Elt < DbgNode->getNumOperands())
109     if (ConstantInt *CI
110         = dyn_cast_or_null<ConstantInt>(DbgNode->getOperand(Elt)))
111       return CI->getSExtValue();
112
113   return 0;
114 }
115
116 DIDescriptor DIDescriptor::getDescriptorField(unsigned Elt) const {
117   if (DbgNode == 0)
118     return DIDescriptor();
119
120   if (Elt < DbgNode->getNumOperands())
121     return
122       DIDescriptor(dyn_cast_or_null<const MDNode>(DbgNode->getOperand(Elt)));
123   return DIDescriptor();
124 }
125
126 GlobalVariable *DIDescriptor::getGlobalVariableField(unsigned Elt) const {
127   if (DbgNode == 0)
128     return 0;
129
130   if (Elt < DbgNode->getNumOperands())
131       return dyn_cast_or_null<GlobalVariable>(DbgNode->getOperand(Elt));
132   return 0;
133 }
134
135 Constant *DIDescriptor::getConstantField(unsigned Elt) const {
136   if (DbgNode == 0)
137     return 0;
138
139   if (Elt < DbgNode->getNumOperands())
140       return dyn_cast_or_null<Constant>(DbgNode->getOperand(Elt));
141   return 0;
142 }
143
144 Function *DIDescriptor::getFunctionField(unsigned Elt) const {
145   if (DbgNode == 0)
146     return 0;
147
148   if (Elt < DbgNode->getNumOperands())
149       return dyn_cast_or_null<Function>(DbgNode->getOperand(Elt));
150   return 0;
151 }
152
153 void DIDescriptor::replaceFunctionField(unsigned Elt, Function *F) {
154   if (DbgNode == 0)
155     return;
156
157   if (Elt < DbgNode->getNumOperands()) {
158     MDNode *Node = const_cast<MDNode*>(DbgNode);
159     Node->replaceOperandWith(Elt, F);
160   }
161 }
162
163 unsigned DIVariable::getNumAddrElements() const {
164   return DbgNode->getNumOperands()-8;
165 }
166
167 /// getInlinedAt - If this variable is inlined then return inline location.
168 MDNode *DIVariable::getInlinedAt() const {
169   return dyn_cast_or_null<MDNode>(DbgNode->getOperand(7));
170 }
171
172 //===----------------------------------------------------------------------===//
173 // Predicates
174 //===----------------------------------------------------------------------===//
175
176 /// isBasicType - Return true if the specified tag is legal for
177 /// DIBasicType.
178 bool DIDescriptor::isBasicType() const {
179   if (!DbgNode) return false;
180   switch (getTag()) {
181   case dwarf::DW_TAG_base_type:
182   case dwarf::DW_TAG_unspecified_type:
183     return true;
184   default:
185     return false;
186   }
187 }
188
189 /// isDerivedType - Return true if the specified tag is legal for DIDerivedType.
190 bool DIDescriptor::isDerivedType() const {
191   if (!DbgNode) return false;
192   switch (getTag()) {
193   case dwarf::DW_TAG_typedef:
194   case dwarf::DW_TAG_pointer_type:
195   case dwarf::DW_TAG_ptr_to_member_type:
196   case dwarf::DW_TAG_reference_type:
197   case dwarf::DW_TAG_rvalue_reference_type:
198   case dwarf::DW_TAG_const_type:
199   case dwarf::DW_TAG_volatile_type:
200   case dwarf::DW_TAG_restrict_type:
201   case dwarf::DW_TAG_member:
202   case dwarf::DW_TAG_inheritance:
203   case dwarf::DW_TAG_friend:
204     return true;
205   default:
206     // CompositeTypes are currently modelled as DerivedTypes.
207     return isCompositeType();
208   }
209 }
210
211 /// isCompositeType - Return true if the specified tag is legal for
212 /// DICompositeType.
213 bool DIDescriptor::isCompositeType() const {
214   if (!DbgNode) return false;
215   switch (getTag()) {
216   case dwarf::DW_TAG_array_type:
217   case dwarf::DW_TAG_structure_type:
218   case dwarf::DW_TAG_union_type:
219   case dwarf::DW_TAG_enumeration_type:
220   case dwarf::DW_TAG_subroutine_type:
221   case dwarf::DW_TAG_class_type:
222     return true;
223   default:
224     return false;
225   }
226 }
227
228 /// isVariable - Return true if the specified tag is legal for DIVariable.
229 bool DIDescriptor::isVariable() const {
230   if (!DbgNode) return false;
231   switch (getTag()) {
232   case dwarf::DW_TAG_auto_variable:
233   case dwarf::DW_TAG_arg_variable:
234     return true;
235   default:
236     return false;
237   }
238 }
239
240 /// isType - Return true if the specified tag is legal for DIType.
241 bool DIDescriptor::isType() const {
242   return isBasicType() || isCompositeType() || isDerivedType();
243 }
244
245 /// isSubprogram - Return true if the specified tag is legal for
246 /// DISubprogram.
247 bool DIDescriptor::isSubprogram() const {
248   return DbgNode && getTag() == dwarf::DW_TAG_subprogram;
249 }
250
251 /// isGlobalVariable - Return true if the specified tag is legal for
252 /// DIGlobalVariable.
253 bool DIDescriptor::isGlobalVariable() const {
254   return DbgNode && (getTag() == dwarf::DW_TAG_variable ||
255                      getTag() == dwarf::DW_TAG_constant);
256 }
257
258 /// isGlobal - Return true if the specified tag is legal for DIGlobal.
259 bool DIDescriptor::isGlobal() const {
260   return isGlobalVariable();
261 }
262
263 /// isUnspecifiedParmeter - Return true if the specified tag is
264 /// DW_TAG_unspecified_parameters.
265 bool DIDescriptor::isUnspecifiedParameter() const {
266   return DbgNode && getTag() == dwarf::DW_TAG_unspecified_parameters;
267 }
268
269 /// isScope - Return true if the specified tag is one of the scope
270 /// related tag.
271 bool DIDescriptor::isScope() const {
272   if (!DbgNode) return false;
273   switch (getTag()) {
274   case dwarf::DW_TAG_compile_unit:
275   case dwarf::DW_TAG_lexical_block:
276   case dwarf::DW_TAG_subprogram:
277   case dwarf::DW_TAG_namespace:
278     return true;
279   default:
280     break;
281   }
282   return false;
283 }
284
285 /// isTemplateTypeParameter - Return true if the specified tag is
286 /// DW_TAG_template_type_parameter.
287 bool DIDescriptor::isTemplateTypeParameter() const {
288   return DbgNode && getTag() == dwarf::DW_TAG_template_type_parameter;
289 }
290
291 /// isTemplateValueParameter - Return true if the specified tag is
292 /// DW_TAG_template_value_parameter.
293 bool DIDescriptor::isTemplateValueParameter() const {
294   return DbgNode && getTag() == dwarf::DW_TAG_template_value_parameter;
295 }
296
297 /// isCompileUnit - Return true if the specified tag is DW_TAG_compile_unit.
298 bool DIDescriptor::isCompileUnit() const {
299   return DbgNode && getTag() == dwarf::DW_TAG_compile_unit;
300 }
301
302 /// isFile - Return true if the specified tag is DW_TAG_file_type.
303 bool DIDescriptor::isFile() const {
304   return DbgNode && getTag() == dwarf::DW_TAG_file_type;
305 }
306
307 /// isNameSpace - Return true if the specified tag is DW_TAG_namespace.
308 bool DIDescriptor::isNameSpace() const {
309   return DbgNode && getTag() == dwarf::DW_TAG_namespace;
310 }
311
312 /// isLexicalBlockFile - Return true if the specified descriptor is a
313 /// lexical block with an extra file.
314 bool DIDescriptor::isLexicalBlockFile() const {
315   return DbgNode && getTag() == dwarf::DW_TAG_lexical_block &&
316     (DbgNode->getNumOperands() == 3);
317 }
318
319 /// isLexicalBlock - Return true if the specified tag is DW_TAG_lexical_block.
320 bool DIDescriptor::isLexicalBlock() const {
321   return DbgNode && getTag() == dwarf::DW_TAG_lexical_block &&
322     (DbgNode->getNumOperands() > 3);
323 }
324
325 /// isSubrange - Return true if the specified tag is DW_TAG_subrange_type.
326 bool DIDescriptor::isSubrange() const {
327   return DbgNode && getTag() == dwarf::DW_TAG_subrange_type;
328 }
329
330 /// isEnumerator - Return true if the specified tag is DW_TAG_enumerator.
331 bool DIDescriptor::isEnumerator() const {
332   return DbgNode && getTag() == dwarf::DW_TAG_enumerator;
333 }
334
335 /// isObjCProperty - Return true if the specified tag is DW_TAG
336 bool DIDescriptor::isObjCProperty() const {
337   return DbgNode && getTag() == dwarf::DW_TAG_APPLE_property;
338 }
339 //===----------------------------------------------------------------------===//
340 // Simple Descriptor Constructors and other Methods
341 //===----------------------------------------------------------------------===//
342
343 DIType::DIType(const MDNode *N) : DIScope(N) {
344   if (!N) return;
345   if (!isBasicType() && !isDerivedType() && !isCompositeType()) {
346     DbgNode = 0;
347   }
348 }
349
350 unsigned DIArray::getNumElements() const {
351   if (!DbgNode)
352     return 0;
353   return DbgNode->getNumOperands();
354 }
355
356 /// replaceAllUsesWith - Replace all uses of debug info referenced by
357 /// this descriptor.
358 void DIType::replaceAllUsesWith(DIDescriptor &D) {
359   if (!DbgNode)
360     return;
361
362   // Since we use a TrackingVH for the node, its easy for clients to manufacture
363   // legitimate situations where they want to replaceAllUsesWith() on something
364   // which, due to uniquing, has merged with the source. We shield clients from
365   // this detail by allowing a value to be replaced with replaceAllUsesWith()
366   // itself.
367   if (DbgNode != D) {
368     MDNode *Node = const_cast<MDNode*>(DbgNode);
369     const MDNode *DN = D;
370     const Value *V = cast_or_null<Value>(DN);
371     Node->replaceAllUsesWith(const_cast<Value*>(V));
372     MDNode::deleteTemporary(Node);
373   }
374 }
375
376 /// replaceAllUsesWith - Replace all uses of debug info referenced by
377 /// this descriptor.
378 void DIType::replaceAllUsesWith(MDNode *D) {
379   if (!DbgNode)
380     return;
381
382   // Since we use a TrackingVH for the node, its easy for clients to manufacture
383   // legitimate situations where they want to replaceAllUsesWith() on something
384   // which, due to uniquing, has merged with the source. We shield clients from
385   // this detail by allowing a value to be replaced with replaceAllUsesWith()
386   // itself.
387   if (DbgNode != D) {
388     MDNode *Node = const_cast<MDNode*>(DbgNode);
389     const MDNode *DN = D;
390     const Value *V = cast_or_null<Value>(DN);
391     Node->replaceAllUsesWith(const_cast<Value*>(V));
392     MDNode::deleteTemporary(Node);
393   }
394 }
395
396 /// isUnsignedDIType - Return true if type encoding is unsigned.
397 bool DIType::isUnsignedDIType() {
398   DIDerivedType DTy(DbgNode);
399   if (DTy.Verify())
400     return DTy.getTypeDerivedFrom().isUnsignedDIType();
401
402   DIBasicType BTy(DbgNode);
403   if (BTy.Verify()) {
404     unsigned Encoding = BTy.getEncoding();
405     if (Encoding == dwarf::DW_ATE_unsigned ||
406         Encoding == dwarf::DW_ATE_unsigned_char ||
407         Encoding == dwarf::DW_ATE_boolean)
408       return true;
409   }
410   return false;
411 }
412
413 /// Verify - Verify that a compile unit is well formed.
414 bool DICompileUnit::Verify() const {
415   if (!isCompileUnit())
416     return false;
417   StringRef N = getFilename();
418   if (N.empty())
419     return false;
420   // It is possible that directory and produce string is empty.
421   return DbgNode->getNumOperands() == 12;
422 }
423
424 /// Verify - Verify that an ObjC property is well formed.
425 bool DIObjCProperty::Verify() const {
426   if (!isObjCProperty())
427     return false;
428
429   DIType Ty = getType();
430   if (!Ty.Verify()) return false;
431
432   // Don't worry about the rest of the strings for now.
433   return DbgNode->getNumOperands() == 8;
434 }
435
436 /// Verify - Verify that a type descriptor is well formed.
437 bool DIType::Verify() const {
438   if (!isType())
439     return false;
440   if (getContext() && !getContext().Verify())
441     return false;
442   unsigned Tag = getTag();
443   if (!isBasicType() && Tag != dwarf::DW_TAG_const_type &&
444       Tag != dwarf::DW_TAG_volatile_type && Tag != dwarf::DW_TAG_pointer_type &&
445       Tag != dwarf::DW_TAG_ptr_to_member_type &&
446       Tag != dwarf::DW_TAG_reference_type &&
447       Tag != dwarf::DW_TAG_rvalue_reference_type &&
448       Tag != dwarf::DW_TAG_restrict_type &&
449       Tag != dwarf::DW_TAG_array_type &&
450       Tag != dwarf::DW_TAG_enumeration_type &&
451       Tag != dwarf::DW_TAG_subroutine_type &&
452       getFilename().empty())
453     return false;
454   return true;
455 }
456
457 /// Verify - Verify that a basic type descriptor is well formed.
458 bool DIBasicType::Verify() const {
459   return isBasicType() && DbgNode->getNumOperands() == 10;
460 }
461
462 /// Verify - Verify that a derived type descriptor is well formed.
463 bool DIDerivedType::Verify() const {
464   return isDerivedType() && DbgNode->getNumOperands() >= 10 &&
465          DbgNode->getNumOperands() <= 14;
466 }
467
468 /// Verify - Verify that a composite type descriptor is well formed.
469 bool DICompositeType::Verify() const {
470   if (!isCompositeType())
471     return false;
472   if (getContext() && !getContext().Verify())
473     return false;
474
475   return DbgNode->getNumOperands() >= 10 && DbgNode->getNumOperands() <= 14;
476 }
477
478 /// Verify - Verify that a subprogram descriptor is well formed.
479 bool DISubprogram::Verify() const {
480   if (!isSubprogram())
481     return false;
482
483   if (getContext() && !getContext().Verify())
484     return false;
485
486   DICompositeType Ty = getType();
487   if (!Ty.Verify())
488     return false;
489   return DbgNode->getNumOperands() == 20;
490 }
491
492 /// Verify - Verify that a global variable descriptor is well formed.
493 bool DIGlobalVariable::Verify() const {
494   if (!isGlobalVariable())
495     return false;
496
497   if (getDisplayName().empty())
498     return false;
499
500   if (getContext() && !getContext().Verify())
501     return false;
502
503   DIType Ty = getType();
504   if (!Ty.Verify())
505     return false;
506
507   if (!getGlobal() && !getConstant())
508     return false;
509
510   return DbgNode->getNumOperands() == 13;
511 }
512
513 /// Verify - Verify that a variable descriptor is well formed.
514 bool DIVariable::Verify() const {
515   if (!isVariable())
516     return false;
517
518   if (getContext() && !getContext().Verify())
519     return false;
520
521   DIType Ty = getType();
522   if (!Ty.Verify())
523     return false;
524
525   return DbgNode->getNumOperands() >= 8;
526 }
527
528 /// Verify - Verify that a location descriptor is well formed.
529 bool DILocation::Verify() const {
530   if (!DbgNode)
531     return false;
532
533   return DbgNode->getNumOperands() == 4;
534 }
535
536 /// Verify - Verify that a namespace descriptor is well formed.
537 bool DINameSpace::Verify() const {
538   if (!isNameSpace())
539     return false;
540   return DbgNode->getNumOperands() == 5;
541 }
542
543 /// \brief Retrieve the MDNode for the directory/file pair.
544 MDNode *DIFile::getFileNode() const {
545   return const_cast<MDNode*>(getNodeField(DbgNode, 1));
546 }
547
548 /// \brief Verify that the file descriptor is well formed.
549 bool DIFile::Verify() const {
550   return isFile() && DbgNode->getNumOperands() == 2;
551 }
552
553 /// \brief Verify that the enumerator descriptor is well formed.
554 bool DIEnumerator::Verify() const {
555   return isEnumerator() && DbgNode->getNumOperands() == 3;
556 }
557
558 /// \brief Verify that the subrange descriptor is well formed.
559 bool DISubrange::Verify() const {
560   return isSubrange() && DbgNode->getNumOperands() == 3;
561 }
562
563 /// \brief Verify that the lexical block descriptor is well formed.
564 bool DILexicalBlock::Verify() const {
565   return isLexicalBlock() && DbgNode->getNumOperands() == 6;
566 }
567
568 /// \brief Verify that the file-scoped lexical block descriptor is well formed.
569 bool DILexicalBlockFile::Verify() const {
570   return isLexicalBlockFile() && DbgNode->getNumOperands() == 3;
571 }
572
573 /// \brief Verify that the template type parameter descriptor is well formed.
574 bool DITemplateTypeParameter::Verify() const {
575   return isTemplateTypeParameter() && DbgNode->getNumOperands() == 7;
576 }
577
578 /// \brief Verify that the template value parameter descriptor is well formed.
579 bool DITemplateValueParameter::Verify() const {
580   return isTemplateValueParameter() && DbgNode->getNumOperands() == 8;
581 }
582
583 /// getOriginalTypeSize - If this type is derived from a base type then
584 /// return base type size.
585 uint64_t DIDerivedType::getOriginalTypeSize() const {
586   unsigned Tag = getTag();
587
588   if (Tag != dwarf::DW_TAG_member && Tag != dwarf::DW_TAG_typedef &&
589       Tag != dwarf::DW_TAG_const_type && Tag != dwarf::DW_TAG_volatile_type &&
590       Tag != dwarf::DW_TAG_restrict_type)
591     return getSizeInBits();
592
593   DIType BaseType = getTypeDerivedFrom();
594
595   // If this type is not derived from any type then take conservative approach.
596   if (!BaseType.isValid())
597     return getSizeInBits();
598
599   // If this is a derived type, go ahead and get the base type, unless it's a
600   // reference then it's just the size of the field. Pointer types have no need
601   // of this since they're a different type of qualification on the type.
602   if (BaseType.getTag() == dwarf::DW_TAG_reference_type ||
603       BaseType.getTag() == dwarf::DW_TAG_rvalue_reference_type)
604     return getSizeInBits();
605
606   if (BaseType.isDerivedType())
607     return DIDerivedType(BaseType).getOriginalTypeSize();
608
609   return BaseType.getSizeInBits();
610 }
611
612 /// getObjCProperty - Return property node, if this ivar is associated with one.
613 MDNode *DIDerivedType::getObjCProperty() const {
614   if (DbgNode->getNumOperands() <= 10)
615     return NULL;
616   return dyn_cast_or_null<MDNode>(DbgNode->getOperand(10));
617 }
618
619 /// \brief Set the array of member DITypes.
620 void DICompositeType::setTypeArray(DIArray Elements, DIArray TParams) {
621   assert(!TParams || DbgNode->getNumOperands() == 14 && "If you're setting the template parameters this should include a slot for that");
622   TrackingVH<MDNode> N(*this);
623   N->replaceOperandWith(10, Elements);
624   if (TParams)
625     N->replaceOperandWith(13, TParams);
626   DbgNode = N;
627 }
628
629 /// \brief Set the containing type.
630 void DICompositeType::setContainingType(DICompositeType ContainingType) {
631   TrackingVH<MDNode> N(*this);
632   N->replaceOperandWith(12, ContainingType);
633   DbgNode = N;
634 }
635
636 /// isInlinedFnArgument - Return true if this variable provides debugging
637 /// information for an inlined function arguments.
638 bool DIVariable::isInlinedFnArgument(const Function *CurFn) {
639   assert(CurFn && "Invalid function");
640   if (!getContext().isSubprogram())
641     return false;
642   // This variable is not inlined function argument if its scope
643   // does not describe current function.
644   return !DISubprogram(getContext()).describes(CurFn);
645 }
646
647 /// describes - Return true if this subprogram provides debugging
648 /// information for the function F.
649 bool DISubprogram::describes(const Function *F) {
650   assert(F && "Invalid function");
651   if (F == getFunction())
652     return true;
653   StringRef Name = getLinkageName();
654   if (Name.empty())
655     Name = getName();
656   if (F->getName() == Name)
657     return true;
658   return false;
659 }
660
661 unsigned DISubprogram::isOptimized() const {
662   assert (DbgNode && "Invalid subprogram descriptor!");
663   if (DbgNode->getNumOperands() == 15)
664     return getUnsignedField(14);
665   return 0;
666 }
667
668 MDNode *DISubprogram::getVariablesNodes() const {
669   if (!DbgNode || DbgNode->getNumOperands() <= 18)
670     return NULL;
671   return dyn_cast_or_null<MDNode>(DbgNode->getOperand(18));
672 }
673
674 DIArray DISubprogram::getVariables() const {
675   if (!DbgNode || DbgNode->getNumOperands() <= 18)
676     return DIArray();
677   if (MDNode *T = dyn_cast_or_null<MDNode>(DbgNode->getOperand(18)))
678     return DIArray(T);
679   return DIArray();
680 }
681
682 StringRef DIScope::getFilename() const {
683   if (!DbgNode)
684     return StringRef();
685   return ::getStringField(getNodeField(DbgNode, 1), 0);
686 }
687
688 StringRef DIScope::getDirectory() const {
689   if (!DbgNode)
690     return StringRef();
691   return ::getStringField(getNodeField(DbgNode, 1), 1);
692 }
693
694 DIArray DICompileUnit::getEnumTypes() const {
695   if (!DbgNode || DbgNode->getNumOperands() < 12)
696     return DIArray();
697
698   if (MDNode *N = dyn_cast_or_null<MDNode>(DbgNode->getOperand(7)))
699     return DIArray(N);
700   return DIArray();
701 }
702
703 DIArray DICompileUnit::getRetainedTypes() const {
704   if (!DbgNode || DbgNode->getNumOperands() < 12)
705     return DIArray();
706
707   if (MDNode *N = dyn_cast_or_null<MDNode>(DbgNode->getOperand(8)))
708     return DIArray(N);
709   return DIArray();
710 }
711
712 DIArray DICompileUnit::getSubprograms() const {
713   if (!DbgNode || DbgNode->getNumOperands() < 12)
714     return DIArray();
715
716   if (MDNode *N = dyn_cast_or_null<MDNode>(DbgNode->getOperand(9)))
717     return DIArray(N);
718   return DIArray();
719 }
720
721
722 DIArray DICompileUnit::getGlobalVariables() const {
723   if (!DbgNode || DbgNode->getNumOperands() < 12)
724     return DIArray();
725
726   if (MDNode *N = dyn_cast_or_null<MDNode>(DbgNode->getOperand(10)))
727     return DIArray(N);
728   return DIArray();
729 }
730
731 /// fixupObjcLikeName - Replace contains special characters used
732 /// in a typical Objective-C names with '.' in a given string.
733 static void fixupObjcLikeName(StringRef Str, SmallVectorImpl<char> &Out) {
734   bool isObjCLike = false;
735   for (size_t i = 0, e = Str.size(); i < e; ++i) {
736     char C = Str[i];
737     if (C == '[')
738       isObjCLike = true;
739
740     if (isObjCLike && (C == '[' || C == ']' || C == ' ' || C == ':' ||
741                        C == '+' || C == '(' || C == ')'))
742       Out.push_back('.');
743     else
744       Out.push_back(C);
745   }
746 }
747
748 /// getFnSpecificMDNode - Return a NameMDNode, if available, that is
749 /// suitable to hold function specific information.
750 NamedMDNode *llvm::getFnSpecificMDNode(const Module &M, DISubprogram Fn) {
751   SmallString<32> Name = StringRef("llvm.dbg.lv.");
752   StringRef FName = "fn";
753   if (Fn.getFunction())
754     FName = Fn.getFunction()->getName();
755   else
756     FName = Fn.getName();
757   char One = '\1';
758   if (FName.startswith(StringRef(&One, 1)))
759     FName = FName.substr(1);
760   fixupObjcLikeName(FName, Name);
761   return M.getNamedMetadata(Name.str());
762 }
763
764 /// getOrInsertFnSpecificMDNode - Return a NameMDNode that is suitable
765 /// to hold function specific information.
766 NamedMDNode *llvm::getOrInsertFnSpecificMDNode(Module &M, DISubprogram Fn) {
767   SmallString<32> Name = StringRef("llvm.dbg.lv.");
768   StringRef FName = "fn";
769   if (Fn.getFunction())
770     FName = Fn.getFunction()->getName();
771   else
772     FName = Fn.getName();
773   char One = '\1';
774   if (FName.startswith(StringRef(&One, 1)))
775     FName = FName.substr(1);
776   fixupObjcLikeName(FName, Name);
777
778   return M.getOrInsertNamedMetadata(Name.str());
779 }
780
781 /// createInlinedVariable - Create a new inlined variable based on current
782 /// variable.
783 /// @param DV            Current Variable.
784 /// @param InlinedScope  Location at current variable is inlined.
785 DIVariable llvm::createInlinedVariable(MDNode *DV, MDNode *InlinedScope,
786                                        LLVMContext &VMContext) {
787   SmallVector<Value *, 16> Elts;
788   // Insert inlined scope as 7th element.
789   for (unsigned i = 0, e = DV->getNumOperands(); i != e; ++i)
790     i == 7 ? Elts.push_back(InlinedScope) :
791              Elts.push_back(DV->getOperand(i));
792   return DIVariable(MDNode::get(VMContext, Elts));
793 }
794
795 /// cleanseInlinedVariable - Remove inlined scope from the variable.
796 DIVariable llvm::cleanseInlinedVariable(MDNode *DV, LLVMContext &VMContext) {
797   SmallVector<Value *, 16> Elts;
798   // Insert inlined scope as 7th element.
799   for (unsigned i = 0, e = DV->getNumOperands(); i != e; ++i)
800     i == 7 ?
801       Elts.push_back(Constant::getNullValue(Type::getInt32Ty(VMContext))):
802       Elts.push_back(DV->getOperand(i));
803   return DIVariable(MDNode::get(VMContext, Elts));
804 }
805
806 /// getDISubprogram - Find subprogram that is enclosing this scope.
807 DISubprogram llvm::getDISubprogram(const MDNode *Scope) {
808   DIDescriptor D(Scope);
809   if (D.isSubprogram())
810     return DISubprogram(Scope);
811
812   if (D.isLexicalBlockFile())
813     return getDISubprogram(DILexicalBlockFile(Scope).getContext());
814
815   if (D.isLexicalBlock())
816     return getDISubprogram(DILexicalBlock(Scope).getContext());
817
818   return DISubprogram();
819 }
820
821 /// getDICompositeType - Find underlying composite type.
822 DICompositeType llvm::getDICompositeType(DIType T) {
823   if (T.isCompositeType())
824     return DICompositeType(T);
825
826   if (T.isDerivedType())
827     return getDICompositeType(DIDerivedType(T).getTypeDerivedFrom());
828
829   return DICompositeType();
830 }
831
832 /// isSubprogramContext - Return true if Context is either a subprogram
833 /// or another context nested inside a subprogram.
834 bool llvm::isSubprogramContext(const MDNode *Context) {
835   if (!Context)
836     return false;
837   DIDescriptor D(Context);
838   if (D.isSubprogram())
839     return true;
840   if (D.isType())
841     return isSubprogramContext(DIType(Context).getContext());
842   return false;
843 }
844
845 //===----------------------------------------------------------------------===//
846 // DebugInfoFinder implementations.
847 //===----------------------------------------------------------------------===//
848
849 /// processModule - Process entire module and collect debug info.
850 void DebugInfoFinder::processModule(const Module &M) {
851   if (NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu")) {
852     for (unsigned i = 0, e = CU_Nodes->getNumOperands(); i != e; ++i) {
853       DICompileUnit CU(CU_Nodes->getOperand(i));
854       addCompileUnit(CU);
855       DIArray GVs = CU.getGlobalVariables();
856       for (unsigned i = 0, e = GVs.getNumElements(); i != e; ++i) {
857         DIGlobalVariable DIG(GVs.getElement(i));
858         if (addGlobalVariable(DIG))
859           processType(DIG.getType());
860       }
861       DIArray SPs = CU.getSubprograms();
862       for (unsigned i = 0, e = SPs.getNumElements(); i != e; ++i)
863         processSubprogram(DISubprogram(SPs.getElement(i)));
864       DIArray EnumTypes = CU.getEnumTypes();
865       for (unsigned i = 0, e = EnumTypes.getNumElements(); i != e; ++i)
866         processType(DIType(EnumTypes.getElement(i)));
867       DIArray RetainedTypes = CU.getRetainedTypes();
868       for (unsigned i = 0, e = RetainedTypes.getNumElements(); i != e; ++i)
869         processType(DIType(RetainedTypes.getElement(i)));
870       // FIXME: We really shouldn't be bailing out after visiting just one CU
871       return;
872     }
873   }
874 }
875
876 /// processLocation - Process DILocation.
877 void DebugInfoFinder::processLocation(DILocation Loc) {
878   if (!Loc.Verify()) return;
879   DIDescriptor S(Loc.getScope());
880   if (S.isCompileUnit())
881     addCompileUnit(DICompileUnit(S));
882   else if (S.isSubprogram())
883     processSubprogram(DISubprogram(S));
884   else if (S.isLexicalBlock())
885     processLexicalBlock(DILexicalBlock(S));
886   else if (S.isLexicalBlockFile()) {
887     DILexicalBlockFile DBF = DILexicalBlockFile(S);
888     processLexicalBlock(DILexicalBlock(DBF.getScope()));
889   }
890   processLocation(Loc.getOrigLocation());
891 }
892
893 /// processType - Process DIType.
894 void DebugInfoFinder::processType(DIType DT) {
895   if (!addType(DT))
896     return;
897   if (DT.isCompositeType()) {
898     DICompositeType DCT(DT);
899     processType(DCT.getTypeDerivedFrom());
900     DIArray DA = DCT.getTypeArray();
901     for (unsigned i = 0, e = DA.getNumElements(); i != e; ++i) {
902       DIDescriptor D = DA.getElement(i);
903       if (D.isType())
904         processType(DIType(D));
905       else if (D.isSubprogram())
906         processSubprogram(DISubprogram(D));
907     }
908   } else if (DT.isDerivedType()) {
909     DIDerivedType DDT(DT);
910     processType(DDT.getTypeDerivedFrom());
911   }
912 }
913
914 /// processLexicalBlock
915 void DebugInfoFinder::processLexicalBlock(DILexicalBlock LB) {
916   DIScope Context = LB.getContext();
917   if (Context.isLexicalBlock())
918     return processLexicalBlock(DILexicalBlock(Context));
919   else if (Context.isLexicalBlockFile()) {
920     DILexicalBlockFile DBF = DILexicalBlockFile(Context);
921     return processLexicalBlock(DILexicalBlock(DBF.getScope()));
922   }
923   else
924     return processSubprogram(DISubprogram(Context));
925 }
926
927 /// processSubprogram - Process DISubprogram.
928 void DebugInfoFinder::processSubprogram(DISubprogram SP) {
929   if (!addSubprogram(SP))
930     return;
931   processType(SP.getType());
932 }
933
934 /// processDeclare - Process DbgDeclareInst.
935 void DebugInfoFinder::processDeclare(const DbgDeclareInst *DDI) {
936   MDNode *N = dyn_cast<MDNode>(DDI->getVariable());
937   if (!N) return;
938
939   DIDescriptor DV(N);
940   if (!DV.isVariable())
941     return;
942
943   if (!NodesSeen.insert(DV))
944     return;
945   processType(DIVariable(N).getType());
946 }
947
948 /// addType - Add type into Tys.
949 bool DebugInfoFinder::addType(DIType DT) {
950   if (!DT.isValid())
951     return false;
952
953   if (!NodesSeen.insert(DT))
954     return false;
955
956   TYs.push_back(DT);
957   return true;
958 }
959
960 /// addCompileUnit - Add compile unit into CUs.
961 bool DebugInfoFinder::addCompileUnit(DICompileUnit CU) {
962   if (!CU.Verify())
963     return false;
964
965   if (!NodesSeen.insert(CU))
966     return false;
967
968   CUs.push_back(CU);
969   return true;
970 }
971
972 /// addGlobalVariable - Add global variable into GVs.
973 bool DebugInfoFinder::addGlobalVariable(DIGlobalVariable DIG) {
974   if (!DIDescriptor(DIG).isGlobalVariable())
975     return false;
976
977   if (!NodesSeen.insert(DIG))
978     return false;
979
980   GVs.push_back(DIG);
981   return true;
982 }
983
984 // addSubprogram - Add subprgoram into SPs.
985 bool DebugInfoFinder::addSubprogram(DISubprogram SP) {
986   if (!DIDescriptor(SP).isSubprogram())
987     return false;
988
989   if (!NodesSeen.insert(SP))
990     return false;
991
992   SPs.push_back(SP);
993   return true;
994 }
995
996 //===----------------------------------------------------------------------===//
997 // DIDescriptor: dump routines for all descriptors.
998 //===----------------------------------------------------------------------===//
999
1000 /// dump - Print descriptor to dbgs() with a newline.
1001 void DIDescriptor::dump() const {
1002   print(dbgs()); dbgs() << '\n';
1003 }
1004
1005 /// print - Print descriptor.
1006 void DIDescriptor::print(raw_ostream &OS) const {
1007   if (!DbgNode) return;
1008
1009   if (const char *Tag = dwarf::TagString(getTag()))
1010     OS << "[ " << Tag << " ]";
1011
1012   if (this->isSubrange()) {
1013     DISubrange(DbgNode).printInternal(OS);
1014   } else if (this->isCompileUnit()) {
1015     DICompileUnit(DbgNode).printInternal(OS);
1016   } else if (this->isFile()) {
1017     DIFile(DbgNode).printInternal(OS);
1018   } else if (this->isEnumerator()) {
1019     DIEnumerator(DbgNode).printInternal(OS);
1020   } else if (this->isBasicType()) {
1021     DIType(DbgNode).printInternal(OS);
1022   } else if (this->isDerivedType()) {
1023     DIDerivedType(DbgNode).printInternal(OS);
1024   } else if (this->isCompositeType()) {
1025     DICompositeType(DbgNode).printInternal(OS);
1026   } else if (this->isSubprogram()) {
1027     DISubprogram(DbgNode).printInternal(OS);
1028   } else if (this->isGlobalVariable()) {
1029     DIGlobalVariable(DbgNode).printInternal(OS);
1030   } else if (this->isVariable()) {
1031     DIVariable(DbgNode).printInternal(OS);
1032   } else if (this->isObjCProperty()) {
1033     DIObjCProperty(DbgNode).printInternal(OS);
1034   } else if (this->isNameSpace()) {
1035     DINameSpace(DbgNode).printInternal(OS);
1036   } else if (this->isScope()) {
1037     DIScope(DbgNode).printInternal(OS);
1038   }
1039 }
1040
1041 void DISubrange::printInternal(raw_ostream &OS) const {
1042   int64_t Count = getCount();
1043   if (Count != -1)
1044     OS << " [" << getLo() << ", " << Count - 1 << ']';
1045   else
1046     OS << " [unbounded]";
1047 }
1048
1049 void DIScope::printInternal(raw_ostream &OS) const {
1050   OS << " [" << getDirectory() << "/" << getFilename() << ']';
1051 }
1052
1053 void DICompileUnit::printInternal(raw_ostream &OS) const {
1054   DIScope::printInternal(OS);
1055   if (const char *Lang = dwarf::LanguageString(getLanguage()))
1056     OS << " [" << Lang << ']';
1057 }
1058
1059 void DIEnumerator::printInternal(raw_ostream &OS) const {
1060   OS << " [" << getName() << " :: " << getEnumValue() << ']';
1061 }
1062
1063 void DIType::printInternal(raw_ostream &OS) const {
1064   if (!DbgNode) return;
1065
1066   StringRef Res = getName();
1067   if (!Res.empty())
1068     OS << " [" << Res << "]";
1069
1070   // TODO: Print context?
1071
1072   OS << " [line " << getLineNumber()
1073      << ", size " << getSizeInBits()
1074      << ", align " << getAlignInBits()
1075      << ", offset " << getOffsetInBits();
1076   if (isBasicType())
1077     if (const char *Enc =
1078         dwarf::AttributeEncodingString(DIBasicType(DbgNode).getEncoding()))
1079       OS << ", enc " << Enc;
1080   OS << "]";
1081
1082   if (isPrivate())
1083     OS << " [private]";
1084   else if (isProtected())
1085     OS << " [protected]";
1086
1087   if (isArtificial())
1088     OS << " [artificial]";
1089
1090   if (isForwardDecl())
1091     OS << " [fwd]";
1092   if (isVector())
1093     OS << " [vector]";
1094   if (isStaticMember())
1095     OS << " [static]";
1096 }
1097
1098 void DIDerivedType::printInternal(raw_ostream &OS) const {
1099   DIType::printInternal(OS);
1100   OS << " [from " << getTypeDerivedFrom().getName() << ']';
1101 }
1102
1103 void DICompositeType::printInternal(raw_ostream &OS) const {
1104   DIType::printInternal(OS);
1105   DIArray A = getTypeArray();
1106   OS << " [" << A.getNumElements() << " elements]";
1107 }
1108
1109 void DINameSpace::printInternal(raw_ostream &OS) const {
1110   StringRef Name = getName();
1111   if (!Name.empty())
1112     OS << " [" << Name << ']';
1113
1114   OS << " [line " << getLineNumber() << ']';
1115 }
1116
1117 void DISubprogram::printInternal(raw_ostream &OS) const {
1118   // TODO : Print context
1119   OS << " [line " << getLineNumber() << ']';
1120
1121   if (isLocalToUnit())
1122     OS << " [local]";
1123
1124   if (isDefinition())
1125     OS << " [def]";
1126
1127   if (getScopeLineNumber() != getLineNumber())
1128     OS << " [scope " << getScopeLineNumber() << "]";
1129
1130   if (isPrivate())
1131     OS << " [private]";
1132   else if (isProtected())
1133     OS << " [protected]";
1134
1135   StringRef Res = getName();
1136   if (!Res.empty())
1137     OS << " [" << Res << ']';
1138 }
1139
1140 void DIGlobalVariable::printInternal(raw_ostream &OS) const {
1141   StringRef Res = getName();
1142   if (!Res.empty())
1143     OS << " [" << Res << ']';
1144
1145   OS << " [line " << getLineNumber() << ']';
1146
1147   // TODO : Print context
1148
1149   if (isLocalToUnit())
1150     OS << " [local]";
1151
1152   if (isDefinition())
1153     OS << " [def]";
1154 }
1155
1156 void DIVariable::printInternal(raw_ostream &OS) const {
1157   StringRef Res = getName();
1158   if (!Res.empty())
1159     OS << " [" << Res << ']';
1160
1161   OS << " [line " << getLineNumber() << ']';
1162 }
1163
1164 void DIObjCProperty::printInternal(raw_ostream &OS) const {
1165   StringRef Name = getObjCPropertyName();
1166   if (!Name.empty())
1167     OS << " [" << Name << ']';
1168
1169   OS << " [line " << getLineNumber()
1170      << ", properties " << getUnsignedField(6) << ']';
1171 }
1172
1173 static void printDebugLoc(DebugLoc DL, raw_ostream &CommentOS,
1174                           const LLVMContext &Ctx) {
1175   if (!DL.isUnknown()) {          // Print source line info.
1176     DIScope Scope(DL.getScope(Ctx));
1177     // Omit the directory, because it's likely to be long and uninteresting.
1178     if (Scope.Verify())
1179       CommentOS << Scope.getFilename();
1180     else
1181       CommentOS << "<unknown>";
1182     CommentOS << ':' << DL.getLine();
1183     if (DL.getCol() != 0)
1184       CommentOS << ':' << DL.getCol();
1185     DebugLoc InlinedAtDL = DebugLoc::getFromDILocation(DL.getInlinedAt(Ctx));
1186     if (!InlinedAtDL.isUnknown()) {
1187       CommentOS << " @[ ";
1188       printDebugLoc(InlinedAtDL, CommentOS, Ctx);
1189       CommentOS << " ]";
1190     }
1191   }
1192 }
1193
1194 void DIVariable::printExtendedName(raw_ostream &OS) const {
1195   const LLVMContext &Ctx = DbgNode->getContext();
1196   StringRef Res = getName();
1197   if (!Res.empty())
1198     OS << Res << "," << getLineNumber();
1199   if (MDNode *InlinedAt = getInlinedAt()) {
1200     DebugLoc InlinedAtDL = DebugLoc::getFromDILocation(InlinedAt);
1201     if (!InlinedAtDL.isUnknown()) {
1202       OS << " @[";
1203       printDebugLoc(InlinedAtDL, OS, Ctx);
1204       OS << "]";
1205     }
1206   }
1207 }