Index: llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFObject.h =================================================================== --- llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFObject.h +++ llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFObject.h @@ -29,6 +29,7 @@ virtual ~DWARFObject() = default; virtual StringRef getFileName() const { llvm_unreachable("unimplemented"); } virtual const object::ObjectFile *getFile() const { return nullptr; } + virtual ArrayRef getSectionNames() const { return {}; } virtual bool isLittleEndian() const = 0; virtual uint8_t getAddressSize() const { llvm_unreachable("unimplemented"); } virtual const DWARFSection &getInfoSection() const { return Dummy; } Index: llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFSection.h =================================================================== --- llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFSection.h +++ llvm/trunk/include/llvm/DebugInfo/DWARF/DWARFSection.h @@ -19,6 +19,11 @@ StringRef Data; }; +struct SectionName { + StringRef Name; + bool IsNameUnique; +}; + } // end namespace llvm #endif // LLVM_DEBUGINFO_DWARF_DWARFSECTION_H Index: llvm/trunk/lib/DebugInfo/DWARF/DWARFContext.cpp =================================================================== --- llvm/trunk/lib/DebugInfo/DWARF/DWARFContext.cpp +++ llvm/trunk/lib/DebugInfo/DWARF/DWARFContext.cpp @@ -934,6 +934,7 @@ uint8_t AddressSize; StringRef FileName; const object::ObjectFile *Obj = nullptr; + std::vector SectionNames; using TypeSectionMap = MapVector>; @@ -1055,9 +1056,13 @@ AddressSize(Obj.getBytesInAddress()), FileName(Obj.getFileName()), Obj(&Obj) { + StringMap SectionAmountMap; for (const SectionRef &Section : Obj.sections()) { StringRef Name; Section.getName(Name); + ++SectionAmountMap[Name]; + SectionNames.push_back({ Name, true }); + // Skip BSS and Virtual sections, they aren't interesting. if (Section.isBSS() || Section.isVirtual()) continue; @@ -1179,6 +1184,10 @@ Map->insert({Reloc.getOffset(), Rel}); } } + + for (SectionName &S : SectionNames) + if (SectionAmountMap[S.Name] > 1) + S.IsNameUnique = false; } Optional find(const DWARFSection &S, @@ -1192,6 +1201,10 @@ const object::ObjectFile *getFile() const override { return Obj; } + ArrayRef getSectionNames() const override { + return SectionNames; + } + bool isLittleEndian() const override { return IsLittleEndian; } StringRef getAbbrevDWOSection() const override { return AbbrevDWOSection; } const DWARFSection &getLineDWOSection() const override { Index: llvm/trunk/lib/DebugInfo/DWARF/DWARFDie.cpp =================================================================== --- llvm/trunk/lib/DebugInfo/DWARF/DWARFDie.cpp +++ llvm/trunk/lib/DebugInfo/DWARF/DWARFDie.cpp @@ -57,18 +57,9 @@ const DWARFAddressRangesVector &Ranges, unsigned AddressSize, unsigned Indent, const DIDumpOptions &DumpOpts) { - StringMap SectionAmountMap; - std::vector SectionNames; - if (Obj.getFile() && !DumpOpts.Brief) { - for (const SectionRef &Section : Obj.getFile()->sections()) { - StringRef Name; - if (Section.getName(Name)) - Name = ""; - - ++SectionAmountMap[Name]; - SectionNames.push_back(Name); - } - } + ArrayRef SectionNames; + if (!DumpOpts.Brief) + SectionNames = Obj.getSectionNames(); for (size_t I = 0; I < Ranges.size(); ++I) { const DWARFAddressRange &R = Ranges[I]; @@ -81,13 +72,11 @@ if (SectionNames.empty() || R.SectionIndex == -1ULL) continue; - StringRef Name = R.SectionIndex < SectionNames.size() - ? SectionNames[R.SectionIndex] - : ""; - OS << format(" \"%s\"", Name.str().c_str()); + StringRef Name = SectionNames[R.SectionIndex].Name; + OS << " \"" << Name << '\"'; - // Print section index if there is more than one section with this name. - if (SectionAmountMap[Name] > 1) + // Print section index if name is not unique. + if (!SectionNames[R.SectionIndex].IsNameUnique) OS << format(" [%u]", R.SectionIndex); } }