From a2f9a2c30d43bfec488d121998ee1f27e5ff54f6 Mon Sep 17 00:00:00 2001 From: David Blaikie Date: Thu, 13 Mar 2014 17:55:28 +0000 Subject: [PATCH] MCDwarf: Refactor line table handling into a single data structure This replaces several "compile unit ID -> thing" mappings in favor of one mapping from CUID to the whole line table structure (files, directories, and lines). This is another step along the way to refactoring out reusable components of line table handling for use when generating debug_line.dwo for fission type units. Also, might be a good basis to fold some of this handling down into MCStreamers to avoid the special case of "One line table when doing asm printing, line table per CU otherwise" by building it into the different MCStreamer implementations. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@203821 91177308-0d34-0410-b5e6-96231b3b80d8 --- include/llvm/MC/MCContext.h | 54 ++++++++++++++++++------------------- include/llvm/MC/MCDwarf.h | 38 +++++++++++++++++++++++++- lib/MC/MCContext.cpp | 12 ++++----- lib/MC/MCDwarf.cpp | 41 +++++++++++++--------------- lib/MC/MCObjectStreamer.cpp | 3 +-- 5 files changed, 88 insertions(+), 60 deletions(-) diff --git a/include/llvm/MC/MCContext.h b/include/llvm/MC/MCContext.h index e93a97a96e6..f0e8f0729b2 100644 --- a/include/llvm/MC/MCContext.h +++ b/include/llvm/MC/MCContext.h @@ -108,9 +108,7 @@ namespace llvm { /// We now emit a line table for each compile unit. To reduce the prologue /// size of each line table, the files and directories used by each compile /// unit are separated. - typedef std::map > MCDwarfFilesMap; - MCDwarfFilesMap MCDwarfFilesCUMap; - std::map > MCDwarfDirsCUMap; + std::map MCDwarfFileTablesCUMap; /// The current dwarf line information from the last dwarf .loc directive. MCDwarfLoc CurrentDwarfLoc; @@ -146,13 +144,8 @@ namespace llvm { /// Darwin). bool AllowTemporaryLabels; - /// The dwarf line information from the .loc directives for the sections - /// with assembled machine instructions have after seeing .loc directives. - std::map MCLineSections; /// The Compile Unit ID that we are currently processing. unsigned DwarfCompileUnitID; - /// The line table start symbol for each Compile Unit. - DenseMap MCLineTableSymbols; void *MachOUniquingMap, *ELFUniquingMap, *COFFUniquingMap; @@ -298,26 +291,38 @@ namespace llvm { bool hasDwarfFiles() const { // Traverse MCDwarfFilesCUMap and check whether each entry is empty. - MCDwarfFilesMap::const_iterator MapB, MapE; - for (MapB = MCDwarfFilesCUMap.begin(), MapE = MCDwarfFilesCUMap.end(); - MapB != MapE; MapB++) - if (!MapB->second.empty()) + for (const auto &FileTable : MCDwarfFileTablesCUMap) + if (!FileTable.second.getMCDwarfFiles().empty()) return true; return false; } + const std::map &getMCDwarfFileTables() const { + return MCDwarfFileTablesCUMap; + } + + MCDwarfFileTable &getMCDwarfFileTable(unsigned CUID) { + return MCDwarfFileTablesCUMap[CUID]; + } + + const MCDwarfFileTable &getMCDwarfFileTable(unsigned CUID) const { + auto I = MCDwarfFileTablesCUMap.find(CUID); + assert(I != MCDwarfFileTablesCUMap.end()); + return I->second; + } + const SmallVectorImpl &getMCDwarfFiles(unsigned CUID = 0) { - return MCDwarfFilesCUMap[CUID]; + return getMCDwarfFileTable(CUID).getMCDwarfFiles(); } const SmallVectorImpl &getMCDwarfDirs(unsigned CUID = 0) { - return MCDwarfDirsCUMap[CUID]; + return getMCDwarfFileTable(CUID).getMCDwarfDirs(); } - const std::map &getMCLineSections() const { - return MCLineSections; - } - std::map &getMCLineSections() { - return MCLineSections; + bool hasMCLineSections() const { + for (const auto &Table : MCDwarfFileTablesCUMap) + if (!Table.second.getMCDwarfFiles().empty() || Table.second.getLabel()) + return true; + return false; } unsigned getDwarfCompileUnitID() { return DwarfCompileUnitID; @@ -325,18 +330,11 @@ namespace llvm { void setDwarfCompileUnitID(unsigned CUIndex) { DwarfCompileUnitID = CUIndex; } - const DenseMap &getMCLineTableSymbols() const { - return MCLineTableSymbols; - } MCSymbol *getMCLineTableSymbol(unsigned ID) const { - DenseMap::const_iterator CIter = - MCLineTableSymbols.find(ID); - if (CIter == MCLineTableSymbols.end()) - return NULL; - return CIter->second; + return getMCDwarfFileTable(ID).getLabel(); } void setMCLineTableSymbol(MCSymbol *Sym, unsigned ID) { - MCLineTableSymbols[ID] = Sym; + getMCDwarfFileTable(ID).setLabel(Sym); } /// setCurrentDwarfLoc - saves the information from the currently parsed diff --git a/include/llvm/MC/MCDwarf.h b/include/llvm/MC/MCDwarf.h index 418c213fcdf..17476cf2b7f 100644 --- a/include/llvm/MC/MCDwarf.h +++ b/include/llvm/MC/MCDwarf.h @@ -200,6 +200,11 @@ public: }; class MCDwarfFileTable { + MCSymbol *Label; + SmallVector MCDwarfDirs; + SmallVector MCDwarfFiles; + MCLineSection MCLineSections; + public: // // This emits the Dwarf file and the line tables for all Compile Units. @@ -208,7 +213,38 @@ public: // // This emits the Dwarf file and the line tables for a given Compile Unit. // - static const MCSymbol *EmitCU(MCStreamer *MCOS, unsigned ID); + const MCSymbol *EmitCU(MCStreamer *MCOS) const; + + const SmallVectorImpl &getMCDwarfDirs() const { + return MCDwarfDirs; + } + + SmallVectorImpl &getMCDwarfDirs() { + return MCDwarfDirs; + } + + const SmallVectorImpl &getMCDwarfFiles() const { + return MCDwarfFiles; + } + + SmallVectorImpl &getMCDwarfFiles() { + return MCDwarfFiles; + } + + const MCLineSection &getMCLineSections() const { + return MCLineSections; + } + MCLineSection &getMCLineSections() { + return MCLineSections; + } + + MCSymbol *getLabel() const { + return Label; + } + + void setLabel(MCSymbol *Label) { + this->Label = Label; + } }; class MCDwarfLineAddr { diff --git a/lib/MC/MCContext.cpp b/lib/MC/MCContext.cpp index 718c3cefaa3..f7e30f5b067 100644 --- a/lib/MC/MCContext.cpp +++ b/lib/MC/MCContext.cpp @@ -84,13 +84,10 @@ void MCContext::reset() { Symbols.clear(); Allocator.Reset(); Instances.clear(); - MCDwarfFilesCUMap.clear(); - MCDwarfDirsCUMap.clear(); + MCDwarfFileTablesCUMap.clear(); MCGenDwarfLabelEntries.clear(); DwarfDebugFlags = StringRef(); - MCLineSections.clear(); DwarfCompileUnitID = 0; - MCLineTableSymbols.clear(); CurrentDwarfLoc = MCDwarfLoc(0,0,0,DWARF2_FLAG_IS_STMT,0,0); // If we have the MachO uniquing map, free it. @@ -337,8 +334,9 @@ unsigned MCContext::GetDwarfFile(StringRef Directory, StringRef FileName, // Note: in GenericAsmParser::ParseDirectiveFile() FileNumber was checked // to not be less than one. This needs to be change to be not less than zero. - SmallVectorImpl& MCDwarfFiles = MCDwarfFilesCUMap[CUID]; - SmallVectorImpl& MCDwarfDirs = MCDwarfDirsCUMap[CUID]; + MCDwarfFileTable &Table = MCDwarfFileTablesCUMap[CUID]; + SmallVectorImpl& MCDwarfFiles = Table.getMCDwarfFiles(); + SmallVectorImpl& MCDwarfDirs = Table.getMCDwarfDirs(); // Make space for this FileNumber in the MCDwarfFiles vector if needed. if (FileNumber >= MCDwarfFiles.size()) { MCDwarfFiles.resize(FileNumber + 1); @@ -399,7 +397,7 @@ unsigned MCContext::GetDwarfFile(StringRef Directory, StringRef FileName, /// isValidDwarfFileNumber - takes a dwarf file number and returns true if it /// currently is assigned and false otherwise. bool MCContext::isValidDwarfFileNumber(unsigned FileNumber, unsigned CUID) { - SmallVectorImpl& MCDwarfFiles = MCDwarfFilesCUMap[CUID]; + const SmallVectorImpl& MCDwarfFiles = getMCDwarfFiles(CUID); if(FileNumber == 0 || FileNumber >= MCDwarfFiles.size()) return false; diff --git a/lib/MC/MCDwarf.cpp b/lib/MC/MCDwarf.cpp index 4bc751fd3d7..dc0601d0a61 100644 --- a/lib/MC/MCDwarf.cpp +++ b/lib/MC/MCDwarf.cpp @@ -82,7 +82,8 @@ void MCLineEntry::Make(MCStreamer *MCOS, const MCSection *Section) { // Add the line entry to this section's entries. MCOS->getContext() - .getMCLineSections()[MCOS->getContext().getDwarfCompileUnitID()] + .getMCDwarfFileTable(MCOS->getContext().getDwarfCompileUnitID()) + .getMCLineSections() .addLineEntry(LineEntry, Section); } @@ -205,27 +206,31 @@ EmitDwarfLineTable(MCStreamer *MCOS, const MCSection *Section, // const MCSymbol *MCDwarfFileTable::Emit(MCStreamer *MCOS) { MCContext &context = MCOS->getContext(); - // Switch to the section where the table will be emitted into. - MCOS->SwitchSection(context.getObjectFileInfo()->getDwarfLineSection()); - const DenseMap &MCLineTableSymbols = - MCOS->getContext().getMCLineTableSymbols(); // CUID and MCLineTableSymbols are set in DwarfDebug, when DwarfDebug does // not exist, CUID will be 0 and MCLineTableSymbols will be empty. // Handle Compile Unit 0, the line table start symbol is the section symbol. - const MCSymbol *LineStartSym = EmitCU(MCOS, 0); + auto I = MCOS->getContext().getMCDwarfFileTables().begin(), + E = MCOS->getContext().getMCDwarfFileTables().end(); + + // Switch to the section where the table will be emitted into. + MCOS->SwitchSection(context.getObjectFileInfo()->getDwarfLineSection()); + + const MCSymbol *LineStartSym = I->second.EmitCU(MCOS); // Handle the rest of the Compile Units. - for (unsigned Is = 1, Ie = MCLineTableSymbols.size(); Is < Ie; Is++) - EmitCU(MCOS, Is); + for (++I; I != E; ++I) + I->second.EmitCU(MCOS); return LineStartSym; } -const MCSymbol *MCDwarfFileTable::EmitCU(MCStreamer *MCOS, unsigned CUID) { +const MCSymbol *MCDwarfFileTable::EmitCU(MCStreamer *MCOS) const { MCContext &context = MCOS->getContext(); + + // Create a symbol at the beginning of the line table. - MCSymbol *LineStartSym = MCOS->getContext().getMCLineTableSymbol(CUID); + MCSymbol *LineStartSym = Label; if (!LineStartSym) LineStartSym = context.CreateTempSymbol(); // Set the value of the symbol, as we are at the start of the line table. @@ -276,8 +281,6 @@ const MCSymbol *MCDwarfFileTable::EmitCU(MCStreamer *MCOS, unsigned CUID) { // Put out the directory and file tables. // First the directory table. - const SmallVectorImpl &MCDwarfDirs = - context.getMCDwarfDirs(CUID); for (unsigned i = 0; i < MCDwarfDirs.size(); i++) { MCOS->EmitBytes(MCDwarfDirs[i]); // the DirectoryName MCOS->EmitBytes(StringRef("\0", 1)); // the null term. of the string @@ -285,8 +288,6 @@ const MCSymbol *MCDwarfFileTable::EmitCU(MCStreamer *MCOS, unsigned CUID) { MCOS->EmitIntValue(0, 1); // Terminate the directory list // Second the file table. - const SmallVectorImpl &MCDwarfFiles = - MCOS->getContext().getMCDwarfFiles(CUID); for (unsigned i = 1; i < MCDwarfFiles.size(); i++) { MCOS->EmitBytes(MCDwarfFiles[i]->getName()); // FileName MCOS->EmitBytes(StringRef("\0", 1)); // the null term. of the string @@ -302,15 +303,11 @@ const MCSymbol *MCDwarfFileTable::EmitCU(MCStreamer *MCOS, unsigned CUID) { MCOS->EmitLabel(ProEndSym); // Put out the line tables. - const std::map &MCLineSections = - MCOS->getContext().getMCLineSections(); - auto Iter = MCLineSections.find(CUID); - if (Iter != MCLineSections.end()) - for (const auto &LineSec : Iter->second.getMCLineEntries()) - EmitDwarfLineTable(MCOS, LineSec.first, LineSec.second); + for (const auto &LineSec : MCLineSections.getMCLineEntries()) + EmitDwarfLineTable(MCOS, LineSec.first, LineSec.second); if (MCOS->getContext().getAsmInfo()->getLinkerRequiresNonEmptyDwarfLines() && - Iter == MCLineSections.end()) { + MCLineSections.getMCLineEntries().empty()) { // The darwin9 linker has a bug (see PR8715). For for 32-bit architectures // it requires: // total_length >= prologue_length + 10 @@ -729,7 +726,7 @@ void MCGenDwarfInfo::Emit(MCStreamer *MCOS, const MCSymbol *LineSectionSymbol) { MCOS->SwitchSection(context.getObjectFileInfo()->getDwarfARangesSection()); // If there are no line table entries then do not emit any section contents. - if (context.getMCLineSections().empty()) + if (!context.hasMCLineSections()) return; // Output the data for .debug_aranges section. diff --git a/lib/MC/MCObjectStreamer.cpp b/lib/MC/MCObjectStreamer.cpp index 119df0a083d..996cf5c09b8 100644 --- a/lib/MC/MCObjectStreamer.cpp +++ b/lib/MC/MCObjectStreamer.cpp @@ -379,8 +379,7 @@ void MCObjectStreamer::EmitZeros(uint64_t NumBytes) { void MCObjectStreamer::FinishImpl() { // Dump out the dwarf file & directory tables and line tables. const MCSymbol *LineSectionSymbol = NULL; - if (!getContext().getMCLineTableSymbols().empty() || - getContext().hasDwarfFiles()) + if (getContext().hasMCLineSections()) LineSectionSymbol = MCDwarfFileTable::Emit(this); // If we are generating dwarf for assembly source files dump out the sections. -- 2.34.1