diff options
Diffstat (limited to 'llvm/lib/MC')
-rw-r--r-- | llvm/lib/MC/ELFObjectWriter.cpp | 5 | ||||
-rw-r--r-- | llvm/lib/MC/MCAssembler.cpp | 21 | ||||
-rw-r--r-- | llvm/lib/MC/MCSection.cpp | 11 | ||||
-rw-r--r-- | llvm/lib/MC/MachObjectWriter.cpp | 21 | ||||
-rw-r--r-- | llvm/lib/MC/WinCOFFObjectWriter.cpp | 9 |
5 files changed, 40 insertions, 27 deletions
diff --git a/llvm/lib/MC/ELFObjectWriter.cpp b/llvm/lib/MC/ELFObjectWriter.cpp index a211dcfecf8..6dd8bd74669 100644 --- a/llvm/lib/MC/ELFObjectWriter.cpp +++ b/llvm/lib/MC/ELFObjectWriter.cpp @@ -1348,8 +1348,9 @@ void ELFObjectWriter::WriteObject(MCAssembler &Asm, SectionOffsetsTy SectionOffsets; std::vector<MCSectionELF *> Groups; std::vector<MCSectionELF *> Relocations; - for (const MCSectionData &SD : Asm) { - const MCSectionELF &Section = static_cast<MCSectionELF &>(SD.getSection()); + for (const MCSection &Sec : Asm) { + const MCSectionELF &Section = static_cast<const MCSectionELF &>(Sec); + const MCSectionData &SD = Section.getSectionData(); uint64_t Padding = OffsetToAlignment(OS.tell(), Section.getAlignment()); WriteZeros(Padding); diff --git a/llvm/lib/MC/MCAssembler.cpp b/llvm/lib/MC/MCAssembler.cpp index 2fa023ee120..fe4e22d2104 100644 --- a/llvm/lib/MC/MCAssembler.cpp +++ b/llvm/lib/MC/MCAssembler.cpp @@ -69,11 +69,11 @@ MCAsmLayout::MCAsmLayout(MCAssembler &Asm) { // Compute the section layout order. Virtual sections must go last. for (MCAssembler::iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it) - if (!it->getSection().isVirtualSection()) - SectionOrder.push_back(&*it); + if (!it->isVirtualSection()) + SectionOrder.push_back(&it->getSectionData()); for (MCAssembler::iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it) - if (it->getSection().isVirtualSection()) - SectionOrder.push_back(&*it); + if (it->isVirtualSection()) + SectionOrder.push_back(&it->getSectionData()); } bool MCAsmLayout::isFragmentValid(const MCFragment *F) const { @@ -290,8 +290,6 @@ MCEncodedFragmentWithFixups::~MCEncodedFragmentWithFixups() { /* *** */ -MCSectionData::MCSectionData() : Section(nullptr) {} - MCSectionData::MCSectionData(MCSection &Section) : Section(&Section) {} MCSectionData::iterator @@ -342,7 +340,6 @@ MCAssembler::~MCAssembler() { void MCAssembler::reset() { Sections.clear(); Symbols.clear(); - SectionMap.clear(); IndirectSymbols.clear(); DataRegions.clear(); LinkerOptions.clear(); @@ -862,9 +859,9 @@ void MCAssembler::Finish() { // Create dummy fragments to eliminate any empty sections, this simplifies // layout. if (it->getFragmentList().empty()) - new MCDataFragment(it); + new MCDataFragment(&it->getSectionData()); - it->getSection().setOrdinal(SectionIndex++); + it->setOrdinal(SectionIndex++); } // Assign layout order indices to sections and fragments. @@ -1084,8 +1081,8 @@ bool MCAssembler::layoutOnce(MCAsmLayout &Layout) { bool WasRelaxed = false; for (iterator it = begin(), ie = end(); it != ie; ++it) { - MCSectionData &SD = *it; - while (layoutSectionOnce(Layout, SD)) + MCSection &Sec = *it; + while (layoutSectionOnce(Layout, Sec.getSectionData())) WasRelaxed = true; } @@ -1261,7 +1258,7 @@ void MCAssembler::dump() { OS << " Sections:[\n "; for (iterator it = begin(), ie = end(); it != ie; ++it) { if (it != begin()) OS << ",\n "; - it->dump(); + it->getSectionData().dump(); } OS << "],\n"; OS << " Symbols:["; diff --git a/llvm/lib/MC/MCSection.cpp b/llvm/lib/MC/MCSection.cpp index 523c53787ce..6ad6496fcb9 100644 --- a/llvm/lib/MC/MCSection.cpp +++ b/llvm/lib/MC/MCSection.cpp @@ -19,6 +19,9 @@ using namespace llvm; // MCSection //===----------------------------------------------------------------------===// +MCSection::MCSection(SectionVariant V, SectionKind K, MCSymbol *Begin) + : Begin(Begin), HasInstructions(false), Data(*this), Variant(V), Kind(K) {} + MCSymbol *MCSection::getEndSymbol(MCContext &Ctx) { if (!End) End = Ctx.createTempSymbol("sec_end", true); @@ -49,6 +52,14 @@ void MCSection::setBundleLockState(BundleLockStateType NewState) { ++BundleLockNestingDepth; } +MCSectionData::iterator MCSection::begin() { return Data.begin(); } + +MCSectionData::iterator MCSection::end() { return Data.end(); } + +MCSectionData::FragmentListType &MCSection::getFragmentList() { + return Data.getFragmentList(); +} + MCSectionData::iterator MCSectionData::begin() { return Fragments.begin(); } MCSectionData::iterator MCSectionData::end() { return Fragments.end(); } diff --git a/llvm/lib/MC/MachObjectWriter.cpp b/llvm/lib/MC/MachObjectWriter.cpp index 79de0f9a936..240aa4da076 100644 --- a/llvm/lib/MC/MachObjectWriter.cpp +++ b/llvm/lib/MC/MachObjectWriter.cpp @@ -540,7 +540,7 @@ void MachObjectWriter::ComputeSymbolTable( unsigned Index = 1; for (MCAssembler::iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it, ++Index) - SectionIndexMap[&it->getSection()] = Index; + SectionIndexMap[&*it] = Index; assert(Index <= 256 && "Too many sections!"); // Build the string table. @@ -622,7 +622,8 @@ void MachObjectWriter::ComputeSymbolTable( for (unsigned i = 0, e = UndefinedSymbolData.size(); i != e; ++i) UndefinedSymbolData[i].Symbol->setIndex(Index++); - for (const MCSectionData &SD : Asm) { + for (const MCSection &Section : Asm) { + const MCSectionData &SD = Section.getSectionData(); std::vector<RelAndSymbol> &Relocs = Relocations[&SD]; for (RelAndSymbol &Rel : Relocs) { if (!Rel.Sym) @@ -801,7 +802,7 @@ void MachObjectWriter::WriteObject(MCAssembler &Asm, uint64_t VMSize = 0; for (MCAssembler::const_iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it) { - const MCSectionData &SD = *it; + const MCSectionData &SD = it->getSectionData(); uint64_t Address = getSectionAddress(&SD); uint64_t Size = Layout.getSectionAddressSize(&SD); uint64_t FileSize = Layout.getSectionFileSize(&SD); @@ -832,10 +833,11 @@ void MachObjectWriter::WriteObject(MCAssembler &Asm, uint64_t RelocTableEnd = SectionDataStart + SectionDataFileSize; for (MCAssembler::const_iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it) { - std::vector<RelAndSymbol> &Relocs = Relocations[it]; + const MCSectionData &SD = it->getSectionData(); + std::vector<RelAndSymbol> &Relocs = Relocations[&SD]; unsigned NumRelocs = Relocs.size(); - uint64_t SectionStart = SectionDataStart + getSectionAddress(it); - WriteSection(Asm, Layout, *it, SectionStart, RelocTableEnd, NumRelocs); + uint64_t SectionStart = SectionDataStart + getSectionAddress(&SD); + WriteSection(Asm, Layout, SD, SectionStart, RelocTableEnd, NumRelocs); RelocTableEnd += NumRelocs * sizeof(MachO::any_relocation_info); } @@ -911,9 +913,10 @@ void MachObjectWriter::WriteObject(MCAssembler &Asm, // Write the actual section data. for (MCAssembler::const_iterator it = Asm.begin(), ie = Asm.end(); it != ie; ++it) { - Asm.writeSectionData(it, Layout); + const MCSectionData &SD = it->getSectionData(); + Asm.writeSectionData(&SD, Layout); - uint64_t Pad = getPaddingSize(it, Layout); + uint64_t Pad = getPaddingSize(&SD, Layout); WriteZeros(Pad); } @@ -925,7 +928,7 @@ void MachObjectWriter::WriteObject(MCAssembler &Asm, ie = Asm.end(); it != ie; ++it) { // Write the section relocation entries, in reverse order to match 'as' // (approximately, the exact algorithm is more complicated than this). - std::vector<RelAndSymbol> &Relocs = Relocations[it]; + std::vector<RelAndSymbol> &Relocs = Relocations[&it->getSectionData()]; for (unsigned i = 0, e = Relocs.size(); i != e; ++i) { Write32(Relocs[e - i - 1].MRE.r_word0); Write32(Relocs[e - i - 1].MRE.r_word1); diff --git a/llvm/lib/MC/WinCOFFObjectWriter.cpp b/llvm/lib/MC/WinCOFFObjectWriter.cpp index 565bca3ee93..3866eb38663 100644 --- a/llvm/lib/MC/WinCOFFObjectWriter.cpp +++ b/llvm/lib/MC/WinCOFFObjectWriter.cpp @@ -636,7 +636,7 @@ void WinCOFFObjectWriter::ExecutePostLayoutBinding(MCAssembler &Asm, // "Define" each section & symbol. This creates section & symbol // entries in the staging area. for (const auto &Section : Asm) - DefineSection(Section); + DefineSection(Section.getSectionData()); for (const MCSymbol &Symbol : Asm.symbols()) if (ExportSymbol(Symbol, Asm)) @@ -946,12 +946,13 @@ void WinCOFFObjectWriter::WriteObject(MCAssembler &Asm, offset += COFF::SectionSize * Header.NumberOfSections; for (const auto &Section : Asm) { - COFFSection *Sec = SectionMap[&Section.getSection()]; + COFFSection *Sec = SectionMap[&Section]; if (Sec->Number == -1) continue; - Sec->Header.SizeOfRawData = Layout.getSectionAddressSize(&Section); + Sec->Header.SizeOfRawData = + Layout.getSectionAddressSize(&Section.getSectionData()); if (IsPhysicalSection(Sec)) { // Align the section data to a four byte boundary. @@ -1035,7 +1036,7 @@ void WinCOFFObjectWriter::WriteObject(MCAssembler &Asm, WriteZeros(SectionDataPadding); - Asm.writeSectionData(j, Layout); + Asm.writeSectionData(&j->getSectionData(), Layout); } if ((*i)->Relocations.size() > 0) { |