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