[yaml2obj][ELF] Allow symbols to reference sections.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@184468 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/Object/ELFYAML.h b/include/llvm/Object/ELFYAML.h
index 29bba0c..b007944 100644 (file)
--- a/include/llvm/Object/ELFYAML.h
+++ b/include/llvm/Object/ELFYAML.h
@@ -57,6 +57,7 @@ struct Symbol {
   StringRef Name;
   ELF_STB Binding;
   ELF_STT Type;
+  StringRef Section;
 };
 struct Section {
   StringRef Name;

diff --git a/lib/Object/ELFYAML.cpp b/lib/Object/ELFYAML.cpp
index 5692f4b..39faf7d 100644 (file)
--- a/lib/Object/ELFYAML.cpp
+++ b/lib/Object/ELFYAML.cpp
@@ -315,6 +315,7 @@ void MappingTraits<ELFYAML::Symbol>::mapping(IO &IO, ELFYAML::Symbol &Symbol) {
   IO.mapOptional("Name", Symbol.Name, StringRef());
   IO.mapOptional("Binding", Symbol.Binding, ELFYAML::ELF_STB(0));
   IO.mapOptional("Type", Symbol.Type, ELFYAML::ELF_STT(0));
+  IO.mapOptional("Section", Symbol.Section, StringRef());
 }
 
 void MappingTraits<ELFYAML::Section>::mapping(IO &IO,

diff --git a/test/Object/yaml2obj-elf-symbol-basic.yaml b/test/Object/yaml2obj-elf-symbol-basic.yaml
index c7593b3..f3883bb 100644 (file)
--- a/test/Object/yaml2obj-elf-symbol-basic.yaml
+++ b/test/Object/yaml2obj-elf-symbol-basic.yaml
@@ -15,6 +15,7 @@ Sections:
     - Name: main
       Binding: STB_GLOBAL
       Type: STT_FUNC
+      Section: .text
 
 # CHECK:      Symbols [
 # CHECK-NEXT:   Symbol {
@@ -23,3 +24,4 @@ Sections:
 # CHECK-NEXT:     Name: main
 # CHECK:          Binding: Global
 # CHECK-NEXT:     Type: Function
+# CHECK:          Section: .text

diff --git a/tools/yaml2obj/yaml2elf.cpp b/tools/yaml2obj/yaml2elf.cpp
index 667718a..421781c 100644 (file)
--- a/tools/yaml2obj/yaml2elf.cpp
+++ b/tools/yaml2obj/yaml2elf.cpp
@@ -199,6 +199,13 @@ static void handleSymtabSectionHeader(
     if (!Sym.Name.empty())
       Symbol.st_name = State.getStringTable().addString(Sym.Name);
     Symbol.setBindingAndType(Sym.Binding, Sym.Type);
+    unsigned Index;
+    if (State.getSN2I().lookupSection(Sym.Section, Index)) {
+      errs() << "error: Unknown section referenced: '" << Sym.Section
+             << "' by YAML symbol " << Sym.Name << ".\n";
+      exit(1);
+    }
+    Symbol.st_shndx = Index;
     Syms.push_back(Symbol);
   }