From c18f66e70dea8abc8c9d86d40088fc8fc8247b20 Mon Sep 17 00:00:00 2001
From: Sean Silva <silvas@purdue.edu>
Date: Thu, 20 Jun 2013 20:59:34 +0000
Subject: [yaml2obj][ELF] Add the section name -> section index map to State.

One of the key things that the YAML format abstracts over is the use of
section numbers for referencing sections. Instead, textual section names
are used, which yaml2obj then translates into appropriate section
numbers. (Technically ELF doesn't care about section names (only section
numbers), but since this is a testing tool, readability counts).

This simplifies using section names as symbolic references in various
parts of the code. An upcoming commit will use this to allow symbols to
reference sections.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@184467 91177308-0d34-0410-b5e6-96231b3b80d8
---
 tools/yaml2obj/yaml2elf.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/tools/yaml2obj/yaml2elf.cpp b/tools/yaml2obj/yaml2elf.cpp
index 501dc2b46f..667718af38 100644
--- a/tools/yaml2obj/yaml2elf.cpp
+++ b/tools/yaml2obj/yaml2elf.cpp
@@ -150,18 +150,21 @@ class ELFState {
   /// \brief The ELF file header.
   Elf_Ehdr &Header;
 
+  SectionNameToIdxMap &SN2I;
+
 public:
 
   ELFState(Elf_Ehdr &Header_, ContiguousBlobAccumulator &Accum,
-           unsigned DotStrtabSecNo_)
+           unsigned DotStrtabSecNo_, SectionNameToIdxMap &SN2I_)
       : DotStrtab(), DotStrtabSecNo(DotStrtabSecNo_),
-        SectionContentAccum(Accum), Header(Header_) {}
+        SectionContentAccum(Accum), Header(Header_), SN2I(SN2I_) {}
 
   unsigned getDotStrTabSecNo() const { return DotStrtabSecNo; }
   StringTableBuilder &getStringTable() { return DotStrtab; }
   ContiguousBlobAccumulator &getSectionContentAccum() {
     return SectionContentAccum;
   }
+  SectionNameToIdxMap &getSN2I() { return SN2I; }
 };
 } // end anonymous namespace
 
@@ -252,8 +255,6 @@ static int writeELF(raw_ostream &OS, const ELFYAML::Object &Doc) {
   const size_t SectionContentBeginOffset =
       Header.e_ehsize + Header.e_shentsize * Header.e_shnum;
   ContiguousBlobAccumulator CBA(SectionContentBeginOffset);
-  ELFState<ELFT> State(Header, CBA, DotStrtabSecNo);
-
   SectionNameToIdxMap SN2I;
   for (unsigned i = 0, e = Sections.size(); i != e; ++i) {
     StringRef Name = Sections[i].Name;
@@ -267,6 +268,8 @@ static int writeELF(raw_ostream &OS, const ELFYAML::Object &Doc) {
     }
   }
 
+  ELFState<ELFT> State(Header, CBA, DotStrtabSecNo, SN2I);
+
   StringTableBuilder SHStrTab;
   std::vector<Elf_Shdr> SHeaders;
   {
-- 
cgit v1.2.3