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