lib/CodeGen/AsmPrinter/DwarfUnit.h

   1 //===-- llvm/CodeGen/DwarfUnit.h - Dwarf Compile Unit ---*- C++ -*--===//
   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 contains support for writing dwarf compile unit.
  11 //
  12 //===----------------------------------------------------------------------===//
  13
  14 #ifndef LLVM_LIB_CODEGEN_ASMPRINTER_DWARFUNIT_H
  15 #define LLVM_LIB_CODEGEN_ASMPRINTER_DWARFUNIT_H
  16
  17 #include "DIE.h"
  18 #include "DwarfDebug.h"
  19 #include "llvm/ADT/DenseMap.h"
  20 #include "llvm/ADT/Optional.h"
  21 #include "llvm/ADT/StringMap.h"
  22 #include "llvm/CodeGen/AsmPrinter.h"
  23 #include "llvm/IR/DIBuilder.h"
  24 #include "llvm/IR/DebugInfo.h"
  25 #include "llvm/MC/MCExpr.h"
  26 #include "llvm/MC/MCSection.h"
  27 #include "llvm/MC/MCDwarf.h"
  28
  29 namespace llvm {
  30
  31 class MachineLocation;
  32 class MachineOperand;
  33 class ConstantInt;
  34 class ConstantFP;
  35 class DbgVariable;
  36 class DwarfCompileUnit;
  37
  38 // Data structure to hold a range for range lists.
  39 class RangeSpan {
  40 public:
  41   RangeSpan(MCSymbol *S, MCSymbol *E) : Start(S), End(E) {}
  42   const MCSymbol *getStart() const { return Start; }
  43   const MCSymbol *getEnd() const { return End; }
  44   void setEnd(const MCSymbol *E) { End = E; }
  45
  46 private:
  47   const MCSymbol *Start, *End;
  48 };
  49
  50 class RangeSpanList {
  51 private:
  52   // Index for locating within the debug_range section this particular span.
  53   MCSymbol *RangeSym;
  54   // List of ranges.
  55   SmallVector<RangeSpan, 2> Ranges;
  56
  57 public:
  58   RangeSpanList(MCSymbol *Sym) : RangeSym(Sym) {}
  59   MCSymbol *getSym() const { return RangeSym; }
  60   const SmallVectorImpl<RangeSpan> &getRanges() const { return Ranges; }
  61   void addRange(RangeSpan Range) { Ranges.push_back(Range); }
  62 };
  63
  64 //===----------------------------------------------------------------------===//
  65 /// Unit - This dwarf writer support class manages information associated
  66 /// with a source file.
  67 class DwarfUnit {
  68 protected:
  69   /// UniqueID - a numeric ID unique among all CUs in the module
  70   unsigned UniqueID;
  71
  72   /// Node - MDNode for the compile unit.
  73   DICompileUnit CUNode;
  74
  75   /// Unit debug information entry.
  76   DIE UnitDie;
  77
  78   /// Offset of the UnitDie from beginning of debug info section.
  79   unsigned DebugInfoOffset;
  80
  81   /// Asm - Target of Dwarf emission.
  82   AsmPrinter *Asm;
  83
  84   // Holders for some common dwarf information.
  85   DwarfDebug *DD;
  86   DwarfFile *DU;
  87
  88   /// IndexTyDie - An anonymous type for index type.  Owned by UnitDie.
  89   DIE *IndexTyDie;
  90
  91   /// MDNodeToDieMap - Tracks the mapping of unit level debug information
  92   /// variables to debug information entries.
  93   DenseMap<const MDNode *, DIE *> MDNodeToDieMap;
  94
  95   /// MDNodeToDIEEntryMap - Tracks the mapping of unit level debug information
  96   /// descriptors to debug information entries using a DIEEntry proxy.
  97   DenseMap<const MDNode *, DIEEntry *> MDNodeToDIEEntryMap;
  98
  99   /// GlobalNames - A map of globally visible named entities for this unit.
 100   StringMap<const DIE *> GlobalNames;
 101
 102   /// GlobalTypes - A map of globally visible types for this unit.
 103   StringMap<const DIE *> GlobalTypes;
 104
 105   /// DIEBlocks - A list of all the DIEBlocks in use.
 106   std::vector<DIEBlock *> DIEBlocks;
 107
 108   /// DIELocs - A list of all the DIELocs in use.
 109   std::vector<DIELoc *> DIELocs;
 110
 111   /// ContainingTypeMap - This map is used to keep track of subprogram DIEs that
 112   /// need DW_AT_containing_type attribute. This attribute points to a DIE that
 113   /// corresponds to the MDNode mapped with the subprogram DIE.
 114   DenseMap<DIE *, const MDNode *> ContainingTypeMap;
 115
 116   // List of ranges for a given compile unit.
 117   SmallVector<RangeSpan, 1> CURanges;
 118
 119   // List of range lists for a given compile unit, separate from the ranges for
 120   // the CU itself.
 121   SmallVector<RangeSpanList, 1> CURangeLists;
 122
 123   // DIEValueAllocator - All DIEValues are allocated through this allocator.
 124   BumpPtrAllocator DIEValueAllocator;
 125
 126   // DIEIntegerOne - A preallocated DIEValue because 1 is used frequently.
 127   DIEInteger *DIEIntegerOne;
 128
 129   /// The section this unit will be emitted in.
 130   const MCSection *Section;
 131
 132   /// A label at the start of the non-dwo section related to this unit.
 133   MCSymbol *SectionSym;
 134
 135   /// The start of the unit within its section.
 136   MCSymbol *LabelBegin;
 137
 138   /// The end of the unit within its section.
 139   MCSymbol *LabelEnd;
 140
 141   /// Skeleton unit associated with this unit.
 142   DwarfUnit *Skeleton;
 143
 144   DwarfUnit(unsigned UID, dwarf::Tag, DICompileUnit CU, AsmPrinter *A,
 145             DwarfDebug *DW, DwarfFile *DWU);
 146
 147 public:
 148   virtual ~DwarfUnit();
 149
 150   /// Set the skeleton unit associated with this unit.
 151   void setSkeleton(DwarfUnit &Skel) { Skeleton = &Skel; }
 152
 153   /// Get the skeleton unit associated with this unit.
 154   DwarfUnit *getSkeleton() const { return Skeleton; }
 155
 156   /// Pass in the SectionSym even though we could recreate it in every compile
 157   /// unit (type units will have actually distinct symbols once they're in
 158   /// comdat sections).
 159   void initSection(const MCSection *Section, MCSymbol *SectionSym) {
 160     assert(!this->Section);
 161     this->Section = Section;
 162     this->SectionSym = SectionSym;
 163     this->LabelBegin =
 164         Asm->GetTempSymbol(Section->getLabelBeginName(), getUniqueID());
 165     this->LabelEnd =
 166         Asm->GetTempSymbol(Section->getLabelEndName(), getUniqueID());
 167   }
 168
 169   const MCSection *getSection() const {
 170     assert(Section);
 171     return Section;
 172   }
 173
 174   /// If there's a skeleton then return the section symbol for the skeleton
 175   /// unit, otherwise return the section symbol for this unit.
 176   MCSymbol *getLocalSectionSym() const {
 177     if (Skeleton)
 178       return Skeleton->getSectionSym();
 179     return getSectionSym();
 180   }
 181
 182   MCSymbol *getSectionSym() const {
 183     assert(Section);
 184     return SectionSym;
 185   }
 186
 187   /// If there's a skeleton then return the begin label for the skeleton unit,
 188   /// otherwise return the local label for this unit.
 189   MCSymbol *getLocalLabelBegin() const {
 190     if (Skeleton)
 191       return Skeleton->getLabelBegin();
 192     return getLabelBegin();
 193   }
 194
 195   MCSymbol *getLabelBegin() const {
 196     assert(Section);
 197     return LabelBegin;
 198   }
 199
 200   MCSymbol *getLabelEnd() const {
 201     assert(Section);
 202     return LabelEnd;
 203   }
 204
 205   // Accessors.
 206   unsigned getUniqueID() const { return UniqueID; }
 207   uint16_t getLanguage() const { return CUNode.getLanguage(); }
 208   DICompileUnit getCUNode() const { return CUNode; }
 209   DIE &getUnitDie() { return UnitDie; }
 210   const StringMap<const DIE *> &getGlobalNames() const { return GlobalNames; }
 211   const StringMap<const DIE *> &getGlobalTypes() const { return GlobalTypes; }
 212
 213   unsigned getDebugInfoOffset() const { return DebugInfoOffset; }
 214   void setDebugInfoOffset(unsigned DbgInfoOff) { DebugInfoOffset = DbgInfoOff; }
 215
 216   /// hasContent - Return true if this compile unit has something to write out.
 217   bool hasContent() const { return !UnitDie.getChildren().empty(); }
 218
 219   /// addRange - Add an address range to the list of ranges for this unit.
 220   void addRange(RangeSpan Range);
 221
 222   /// getRanges - Get the list of ranges for this unit.
 223   const SmallVectorImpl<RangeSpan> &getRanges() const { return CURanges; }
 224   SmallVectorImpl<RangeSpan> &getRanges() { return CURanges; }
 225
 226   /// addRangeList - Add an address range list to the list of range lists.
 227   void addRangeList(RangeSpanList Ranges) { CURangeLists.push_back(Ranges); }
 228
 229   /// getRangeLists - Get the vector of range lists.
 230   const SmallVectorImpl<RangeSpanList> &getRangeLists() const {
 231     return CURangeLists;
 232   }
 233   SmallVectorImpl<RangeSpanList> &getRangeLists() { return CURangeLists; }
 234
 235   /// getParentContextString - Get a string containing the language specific
 236   /// context for a global name.
 237   std::string getParentContextString(DIScope Context) const;
 238
 239   /// addGlobalName - Add a new global entity to the compile unit.
 240   ///
 241   void addGlobalName(StringRef Name, DIE &Die, DIScope Context);
 242
 243   /// addAccelNamespace - Add a new name to the namespace accelerator table.
 244   void addAccelNamespace(StringRef Name, const DIE &Die);
 245
 246   /// getDIE - Returns the debug information entry map slot for the
 247   /// specified debug variable. We delegate the request to DwarfDebug
 248   /// when the MDNode can be part of the type system, since DIEs for
 249   /// the type system can be shared across CUs and the mappings are
 250   /// kept in DwarfDebug.
 251   DIE *getDIE(DIDescriptor D) const;
 252
 253   /// getDIELoc - Returns a fresh newly allocated DIELoc.
 254   DIELoc *getDIELoc() { return new (DIEValueAllocator) DIELoc(); }
 255
 256   /// insertDIE - Insert DIE into the map. We delegate the request to DwarfDebug
 257   /// when the MDNode can be part of the type system, since DIEs for
 258   /// the type system can be shared across CUs and the mappings are
 259   /// kept in DwarfDebug.
 260   void insertDIE(DIDescriptor Desc, DIE *D);
 261
 262   /// addFlag - Add a flag that is true to the DIE.
 263   void addFlag(DIE &Die, dwarf::Attribute Attribute);
 264
 265   /// addUInt - Add an unsigned integer attribute data and value.
 266   void addUInt(DIE &Die, dwarf::Attribute Attribute, Optional<dwarf::Form> Form,
 267                uint64_t Integer);
 268
 269   void addUInt(DIE &Block, dwarf::Form Form, uint64_t Integer);
 270
 271   /// addSInt - Add an signed integer attribute data and value.
 272   void addSInt(DIE &Die, dwarf::Attribute Attribute, Optional<dwarf::Form> Form,
 273                int64_t Integer);
 274
 275   void addSInt(DIELoc &Die, Optional<dwarf::Form> Form, int64_t Integer);
 276
 277   /// addString - Add a string attribute data and value.
 278   void addString(DIE &Die, dwarf::Attribute Attribute, StringRef Str);
 279
 280   /// addLocalString - Add a string attribute data and value.
 281   void addLocalString(DIE &Die, dwarf::Attribute Attribute,
 282                       StringRef Str);
 283
 284   /// addExpr - Add a Dwarf expression attribute data and value.
 285   void addExpr(DIELoc &Die, dwarf::Form Form, const MCExpr *Expr);
 286
 287   /// addLabel - Add a Dwarf label attribute data and value.
 288   void addLabel(DIE &Die, dwarf::Attribute Attribute, dwarf::Form Form,
 289                 const MCSymbol *Label);
 290
 291   void addLabel(DIELoc &Die, dwarf::Form Form, const MCSymbol *Label);
 292
 293   /// addLocationList - Add a Dwarf loclistptr attribute data and value.
 294   void addLocationList(DIE &Die, dwarf::Attribute Attribute, unsigned Index);
 295
 296   /// addSectionLabel - Add a Dwarf section label attribute data and value.
 297   ///
 298   void addSectionLabel(DIE &Die, dwarf::Attribute Attribute,
 299                        const MCSymbol *Label);
 300
 301   /// addSectionOffset - Add an offset into a section attribute data and value.
 302   ///
 303   void addSectionOffset(DIE &Die, dwarf::Attribute Attribute, uint64_t Integer);
 304
 305   /// addOpAddress - Add a dwarf op address data and value using the
 306   /// form given and an op of either DW_FORM_addr or DW_FORM_GNU_addr_index.
 307   void addOpAddress(DIELoc &Die, const MCSymbol *Label);
 308
 309   /// addSectionDelta - Add a label delta attribute data and value.
 310   void addSectionDelta(DIE &Die, dwarf::Attribute Attribute, const MCSymbol *Hi,
 311                        const MCSymbol *Lo);
 312
 313   /// addLabelDelta - Add a label delta attribute data and value.
 314   void addLabelDelta(DIE &Die, dwarf::Attribute Attribute, const MCSymbol *Hi,
 315                      const MCSymbol *Lo);
 316
 317   /// addDIEEntry - Add a DIE attribute data and value.
 318   void addDIEEntry(DIE &Die, dwarf::Attribute Attribute, DIE &Entry);
 319
 320   /// addDIEEntry - Add a DIE attribute data and value.
 321   void addDIEEntry(DIE &Die, dwarf::Attribute Attribute, DIEEntry *Entry);
 322
 323   void addDIETypeSignature(DIE &Die, const DwarfTypeUnit &Type);
 324
 325   /// addBlock - Add block data.
 326   void addBlock(DIE &Die, dwarf::Attribute Attribute, DIELoc *Block);
 327
 328   /// addBlock - Add block data.
 329   void addBlock(DIE &Die, dwarf::Attribute Attribute, DIEBlock *Block);
 330
 331   /// addSourceLine - Add location information to specified debug information
 332   /// entry.
 333   void addSourceLine(DIE &Die, unsigned Line, StringRef File,
 334                      StringRef Directory);
 335   void addSourceLine(DIE &Die, DIVariable V);
 336   void addSourceLine(DIE &Die, DIGlobalVariable G);
 337   void addSourceLine(DIE &Die, DISubprogram SP);
 338   void addSourceLine(DIE &Die, DIType Ty);
 339   void addSourceLine(DIE &Die, DINameSpace NS);
 340   void addSourceLine(DIE &Die, DIObjCProperty Ty);
 341
 342   /// addAddress - Add an address attribute to a die based on the location
 343   /// provided.
 344   void addAddress(DIE &Die, dwarf::Attribute Attribute,
 345                   const MachineLocation &Location, bool Indirect = false);
 346
 347   /// addConstantValue - Add constant value entry in variable DIE.
 348   void addConstantValue(DIE &Die, const MachineOperand &MO, DIType Ty);
 349   void addConstantValue(DIE &Die, const ConstantInt *CI, DIType Ty);
 350   void addConstantValue(DIE &Die, const APInt &Val, DIType Ty);
 351   void addConstantValue(DIE &Die, const APInt &Val, bool Unsigned);
 352   void addConstantValue(DIE &Die, bool Unsigned, uint64_t Val);
 353
 354   /// addConstantFPValue - Add constant value entry in variable DIE.
 355   void addConstantFPValue(DIE &Die, const MachineOperand &MO);
 356   void addConstantFPValue(DIE &Die, const ConstantFP *CFP);
 357
 358   /// addTemplateParams - Add template parameters in buffer.
 359   void addTemplateParams(DIE &Buffer, DIArray TParams);
 360
 361   /// addRegisterOp - Add register operand.
 362   void addRegisterOpPiece(DIELoc &TheDie, unsigned Reg,
 363                           unsigned SizeInBits = 0, unsigned OffsetInBits = 0);
 364
 365   /// addRegisterOffset - Add register offset.
 366   void addRegisterOffset(DIELoc &TheDie, unsigned Reg, int64_t Offset);
 367
 368   /// addComplexAddress - Start with the address based on the location provided,
 369   /// and generate the DWARF information necessary to find the actual variable
 370   /// (navigating the extra location information encoded in the type) based on
 371   /// the starting location.  Add the DWARF information to the die.
 372   void addComplexAddress(const DbgVariable &DV, DIE &Die,
 373                          dwarf::Attribute Attribute,
 374                          const MachineLocation &Location);
 375
 376   // FIXME: Should be reformulated in terms of addComplexAddress.
 377   /// addBlockByrefAddress - Start with the address based on the location
 378   /// provided, and generate the DWARF information necessary to find the
 379   /// actual Block variable (navigating the Block struct) based on the
 380   /// starting location.  Add the DWARF information to the die.  Obsolete,
 381   /// please use addComplexAddress instead.
 382   void addBlockByrefAddress(const DbgVariable &DV, DIE &Die,
 383                             dwarf::Attribute Attribute,
 384                             const MachineLocation &Location);
 385
 386   /// addVariableAddress - Add DW_AT_location attribute for a
 387   /// DbgVariable based on provided MachineLocation.
 388   void addVariableAddress(const DbgVariable &DV, DIE &Die,
 389                           MachineLocation Location);
 390
 391   /// addType - Add a new type attribute to the specified entity. This takes
 392   /// and attribute parameter because DW_AT_friend attributes are also
 393   /// type references.
 394   void addType(DIE &Entity, DIType Ty,
 395                dwarf::Attribute Attribute = dwarf::DW_AT_type);
 396
 397   /// getOrCreateNameSpace - Create a DIE for DINameSpace.
 398   DIE *getOrCreateNameSpace(DINameSpace NS);
 399
 400   /// getOrCreateSubprogramDIE - Create new DIE using SP.
 401   DIE *getOrCreateSubprogramDIE(DISubprogram SP);
 402
 403   void applySubprogramAttributes(DISubprogram SP, DIE &SPDie);
 404   void applySubprogramAttributesToDefinition(DISubprogram SP, DIE &SPDie);
 405   void applyVariableAttributes(const DbgVariable &Var, DIE &VariableDie);
 406
 407   /// getOrCreateTypeDIE - Find existing DIE or create new DIE for the
 408   /// given DIType.
 409   DIE *getOrCreateTypeDIE(const MDNode *N);
 410
 411   /// getOrCreateContextDIE - Get context owner's DIE.
 412   DIE *createTypeDIE(DICompositeType Ty);
 413
 414   /// getOrCreateContextDIE - Get context owner's DIE.
 415   DIE *getOrCreateContextDIE(DIScope Context);
 416
 417   /// constructContainingTypeDIEs - Construct DIEs for types that contain
 418   /// vtables.
 419   void constructContainingTypeDIEs();
 420
 421   /// constructVariableDIE - Construct a DIE for the given DbgVariable.
 422   std::unique_ptr<DIE> constructVariableDIE(DbgVariable &DV,
 423                                             bool Abstract = false);
 424
 425   /// constructSubprogramArguments - Construct function argument DIEs.
 426   void constructSubprogramArguments(DIE &Buffer, DITypeArray Args);
 427
 428   /// Create a DIE with the given Tag, add the DIE to its parent, and
 429   /// call insertDIE if MD is not null.
 430   DIE &createAndAddDIE(unsigned Tag, DIE &Parent,
 431                        DIDescriptor N = DIDescriptor());
 432
 433   /// Compute the size of a header for this unit, not including the initial
 434   /// length field.
 435   virtual unsigned getHeaderSize() const {
 436     return sizeof(int16_t) + // DWARF version number
 437            sizeof(int32_t) + // Offset Into Abbrev. Section
 438            sizeof(int8_t);   // Pointer Size (in bytes)
 439   }
 440
 441   /// Emit the header for this unit, not including the initial length field.
 442   virtual void emitHeader(const MCSymbol *ASectionSym) const;
 443
 444   virtual DwarfCompileUnit &getCU() = 0;
 445
 446   /// constructTypeDIE - Construct type DIE from DICompositeType.
 447   void constructTypeDIE(DIE &Buffer, DICompositeType CTy);
 448
 449 protected:
 450   /// getOrCreateStaticMemberDIE - Create new static data member DIE.
 451   DIE *getOrCreateStaticMemberDIE(DIDerivedType DT);
 452
 453   /// Look up the source ID with the given directory and source file names. If
 454   /// none currently exists, create a new ID and insert it in the line table.
 455   virtual unsigned getOrCreateSourceID(StringRef File, StringRef Directory) = 0;
 456
 457 private:
 458   /// \brief Construct a DIE for the given DbgVariable without initializing the
 459   /// DbgVariable's DIE reference.
 460   std::unique_ptr<DIE> constructVariableDIEImpl(const DbgVariable &DV,
 461                                                 bool Abstract);
 462
 463   /// constructTypeDIE - Construct basic type die from DIBasicType.
 464   void constructTypeDIE(DIE &Buffer, DIBasicType BTy);
 465
 466   /// constructTypeDIE - Construct derived type die from DIDerivedType.
 467   void constructTypeDIE(DIE &Buffer, DIDerivedType DTy);
 468
 469   /// constructSubrangeDIE - Construct subrange DIE from DISubrange.
 470   void constructSubrangeDIE(DIE &Buffer, DISubrange SR, DIE *IndexTy);
 471
 472   /// constructArrayTypeDIE - Construct array type DIE from DICompositeType.
 473   void constructArrayTypeDIE(DIE &Buffer, DICompositeType CTy);
 474
 475   /// constructEnumTypeDIE - Construct enum type DIE from DIEnumerator.
 476   void constructEnumTypeDIE(DIE &Buffer, DICompositeType CTy);
 477
 478   /// constructMemberDIE - Construct member DIE from DIDerivedType.
 479   void constructMemberDIE(DIE &Buffer, DIDerivedType DT);
 480
 481   /// constructTemplateTypeParameterDIE - Construct new DIE for the given
 482   /// DITemplateTypeParameter.
 483   void constructTemplateTypeParameterDIE(DIE &Buffer,
 484                                          DITemplateTypeParameter TP);
 485
 486   /// constructTemplateValueParameterDIE - Construct new DIE for the given
 487   /// DITemplateValueParameter.
 488   void constructTemplateValueParameterDIE(DIE &Buffer,
 489                                           DITemplateValueParameter TVP);
 490
 491   /// getLowerBoundDefault - Return the default lower bound for an array. If the
 492   /// DWARF version doesn't handle the language, return -1.
 493   int64_t getDefaultLowerBound() const;
 494
 495   /// getDIEEntry - Returns the debug information entry for the specified
 496   /// debug variable.
 497   DIEEntry *getDIEEntry(const MDNode *N) const {
 498     return MDNodeToDIEEntryMap.lookup(N);
 499   }
 500
 501   /// insertDIEEntry - Insert debug information entry into the map.
 502   void insertDIEEntry(const MDNode *N, DIEEntry *E) {
 503     MDNodeToDIEEntryMap.insert(std::make_pair(N, E));
 504   }
 505
 506   // getIndexTyDie - Get an anonymous type for index type.
 507   DIE *getIndexTyDie() { return IndexTyDie; }
 508
 509   // setIndexTyDie - Set D as anonymous type for index which can be reused
 510   // later.
 511   void setIndexTyDie(DIE *D) { IndexTyDie = D; }
 512
 513   /// createDIEEntry - Creates a new DIEEntry to be a proxy for a debug
 514   /// information entry.
 515   DIEEntry *createDIEEntry(DIE &Entry);
 516
 517   /// resolve - Look in the DwarfDebug map for the MDNode that
 518   /// corresponds to the reference.
 519   template <typename T> T resolve(DIRef<T> Ref) const {
 520     return DD->resolve(Ref);
 521   }
 522
 523   /// If this is a named finished type then include it in the list of types for
 524   /// the accelerator tables.
 525   void updateAcceleratorTables(DIScope Context, DIType Ty, const DIE &TyDIE);
 526 };
 527
 528 class DwarfCompileUnit : public DwarfUnit {
 529   /// The attribute index of DW_AT_stmt_list in the compile unit DIE, avoiding
 530   /// the need to search for it in applyStmtList.
 531   unsigned stmtListIndex;
 532
 533 public:
 534   DwarfCompileUnit(unsigned UID, DICompileUnit Node, AsmPrinter *A,
 535                    DwarfDebug *DW, DwarfFile *DWU);
 536
 537   void initStmtList(MCSymbol *DwarfLineSectionSym);
 538
 539   /// Apply the DW_AT_stmt_list from this compile unit to the specified DIE.
 540   void applyStmtList(DIE &D);
 541
 542   /// createGlobalVariableDIE - create global variable DIE.
 543   void createGlobalVariableDIE(DIGlobalVariable GV);
 544
 545   /// addLabelAddress - Add a dwarf label attribute data and value using
 546   /// either DW_FORM_addr or DW_FORM_GNU_addr_index.
 547   void addLabelAddress(DIE &Die, dwarf::Attribute Attribute,
 548                        const MCSymbol *Label);
 549
 550   /// addLocalLabelAddress - Add a dwarf label attribute data and value using
 551   /// DW_FORM_addr only.
 552   void addLocalLabelAddress(DIE &Die, dwarf::Attribute Attribute,
 553                             const MCSymbol *Label);
 554
 555   DwarfCompileUnit &getCU() override { return *this; }
 556
 557   unsigned getOrCreateSourceID(StringRef FileName, StringRef DirName) override;
 558 };
 559
 560 class DwarfTypeUnit : public DwarfUnit {
 561 private:
 562   uint64_t TypeSignature;
 563   const DIE *Ty;
 564   DwarfCompileUnit &CU;
 565   MCDwarfDwoLineTable *SplitLineTable;
 566
 567 public:
 568   DwarfTypeUnit(unsigned UID, DwarfCompileUnit &CU, AsmPrinter *A,
 569                 DwarfDebug *DW, DwarfFile *DWU,
 570                 MCDwarfDwoLineTable *SplitLineTable = nullptr);
 571
 572   void setTypeSignature(uint64_t Signature) { TypeSignature = Signature; }
 573   uint64_t getTypeSignature() const { return TypeSignature; }
 574   void setType(const DIE *Ty) { this->Ty = Ty; }
 575
 576   /// Emit the header for this unit, not including the initial length field.
 577   void emitHeader(const MCSymbol *ASectionSym) const override;
 578   unsigned getHeaderSize() const override {
 579     return DwarfUnit::getHeaderSize() + sizeof(uint64_t) + // Type Signature
 580            sizeof(uint32_t);                               // Type DIE Offset
 581   }
 582   void initSection(const MCSection *Section);
 583   // Bring in the base function (taking two args, including the section symbol)
 584   // for use when building DWO type units (they don't go in unique comdat
 585   // sections)
 586   using DwarfUnit::initSection;
 587   DwarfCompileUnit &getCU() override { return CU; }
 588
 589 protected:
 590   unsigned getOrCreateSourceID(StringRef File, StringRef Directory) override;
 591 };
 592 } // end llvm namespace
 593 #endif