commit ecd5f4062f35f2d6d42b587f28ca8dd8001ecaf3
author Michael J. Spencer <bigcheesegs@gmail.com> Thu Jan 10 22:41:42 2013 +0000
committer Michael J. Spencer <bigcheesegs@gmail.com> Thu Jan 10 22:41:42 2013 +0000
tree c530ac2d6d2f8d97fad70de166493aaef23528b2
parent 36b9b8f9451144250bbedfe408ad17e08be775ff

[ELF] Add support for __init_array_{start,end}.

With this change c++ global constructors and destructors work.

llvm-svn: 172133

lld/lib/ReaderWriter/ELF/WriterELF.cpp

diff --git a/lld/lib/ReaderWriter/ELF/WriterELF.cpp b/lld/lib/ReaderWriter/ELF/WriterELF.cpp
index 6c1a633..467db99c 100644
--- a/lld/lib/ReaderWriter/ELF/WriterELF.cpp
+++ b/lld/lib/ReaderWriter/ELF/WriterELF.cpp
@@ -1457,6 +1457,8 @@
     ORDER_EH_FRAMEHDR,
     ORDER_CTORS,
     ORDER_DTORS,
+    ORDER_INIT_ARRAY,
+    ORDER_FINI_ARRAY,
     ORDER_DYNAMIC,
     ORDER_GOT,
     ORDER_GOT_PLT,
@@ -1565,7 +1567,9 @@
       return ORDER_RODATA;
 
     case DefinedAtom::typeData:
-      return ORDER_DATA;
+      return llvm::StringSwitch<Reference::Kind>(name)
+        .StartsWith(".init_array", ORDER_INIT_ARRAY)
+        .Default(ORDER_DATA);
 
     case DefinedAtom::typeZeroFill:
       return ORDER_BSS;
@@ -1624,6 +1628,8 @@
     case ORDER_GOT_PLT:
     case ORDER_DATA:
     case ORDER_BSS:
+    case ORDER_INIT_ARRAY:
+    case ORDER_FINI_ARRAY:
       return llvm::ELF::PT_LOAD;
 
     default:
@@ -1654,6 +1660,8 @@
     case ORDER_GOT:
     case ORDER_GOT_PLT:
     case ORDER_DATA:
+    case ORDER_INIT_ARRAY:
+    case ORDER_FINI_ARRAY:
     case ORDER_BSS:
       return true;
 
@@ -1708,6 +1716,25 @@
     return error_code::success();
   }
 
+  /// \biref Find an output Section given a section name.
+  ///
+  /// \todo Make this not O(n). We can't use _mergedSectionMap because it
+  /// doesn't get virtual addresses set :(
+  Chunk<target_endianness, max_align, is64Bits> *
+  findOutputSection(StringRef name) {
+    for (auto seg : _segments) {
+      for (auto sliceI = seg->slices_begin(),
+                sliceE = seg->slices_end(); sliceI != sliceE; ++sliceI) {
+        for (auto secI = (*sliceI)->sections_begin(),
+                  secE = (*sliceI)->sections_end(); secI != secE; ++secI) {
+          if ((*secI)->name() == name)
+            return *secI;
+        }
+      }
+    }
+    return nullptr;
+  }
+
   /// \brief find a absolute atom given a name
   AbsoluteAtomIterT findAbsoluteAtom(const StringRef name) {
     return std::find_if(_absoluteAtoms.begin(), _absoluteAtoms.end(),
@@ -2211,6 +2238,8 @@
   _runtimeFile.addAbsoluteAtom("__bss_end");
   _runtimeFile.addAbsoluteAtom("_end");
   _runtimeFile.addAbsoluteAtom("end");
+  _runtimeFile.addAbsoluteAtom("__init_array_start");
+  _runtimeFile.addAbsoluteAtom("__init_array_end");
 }
 
 /// \brief Hook in lld to add CRuntime file 
@@ -2234,6 +2263,18 @@
  auto bssEndAtomIter = _layout->findAbsoluteAtom("__bss_end");
  auto underScoreEndAtomIter = _layout->findAbsoluteAtom("_end");
  auto endAtomIter = _layout->findAbsoluteAtom("end");
+ auto initArrayStartIter = _layout->findAbsoluteAtom("__init_array_start");
+ auto initArrayEndIter = _layout->findAbsoluteAtom("__init_array_end");
+
+ auto section = _layout->findOutputSection(".init_array");
+ if (section) {
+   initArrayStartIter->setValue(section->virtualAddr());
+   initArrayEndIter->setValue(section->virtualAddr() +
+                              section->memSize());
+ } else {
+   initArrayStartIter->setValue(0);
+   initArrayEndIter->setValue(0);
+ }
 
  assert(!(bssStartAtomIter == _layout->absAtomsEnd() ||
          bssEndAtomIter == _layout->absAtomsEnd() ||