Index: include/llvm/DebugInfo/DWARF/DWARFObject.h =================================================================== --- include/llvm/DebugInfo/DWARF/DWARFObject.h +++ 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: include/llvm/DebugInfo/DWARF/DWARFSection.h =================================================================== --- include/llvm/DebugInfo/DWARF/DWARFSection.h +++ 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: lib/DebugInfo/DWARF/DWARFContext.cpp =================================================================== --- lib/DebugInfo/DWARF/DWARFContext.cpp +++ 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>; @@ -995,6 +996,21 @@ SmallVector, 4> UncompressedSections; + void collectSectionNames() { + StringMap SectionAmountMap; + for (const SectionRef &Section : Obj->sections()) { + StringRef Name; + if (Section.getName(Name)) + Name = ""; + ++SectionAmountMap[Name]; + SectionNames.push_back({Name, true}); + } + + for (SectionName &S : SectionNames) + if (SectionAmountMap[S.Name] > 1) + S.IsNameUnique = false; + } + StringRef *mapSectionToMember(StringRef Name) { if (DWARFSection *Sec = mapNameToDWARFSection(Name)) return &Sec->Data; @@ -1192,6 +1208,12 @@ const object::ObjectFile *getFile() const override { return Obj; } + ArrayRef getSectionNames() const override { + if (Obj && SectionNames.empty()) + const_cast(this)->collectSectionNames(); + return SectionNames; + } + bool isLittleEndian() const override { return IsLittleEndian; } StringRef getAbbrevDWOSection() const override { return AbbrevDWOSection; } const DWARFSection &getLineDWOSection() const override { Index: lib/DebugInfo/DWARF/DWARFDie.cpp =================================================================== --- lib/DebugInfo/DWARF/DWARFDie.cpp +++ 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); } }