[lld][ELF] Add modified writer

llvm-svn: 176425

1 files changed, 38 insertions, 0 deletions

diff --git a/lld/lib/ReaderWriter/ELF/Writer.cpp b/lld/lib/ReaderWriter/ELF/Writer.cpp
new file mode 100644
index 00000000000..49c7b112388
--- /dev/null
+++ b/lld/lib/ReaderWriter/ELF/Writer.cpp
@@ -0,0 +1,38 @@
+//===- lib/ReaderWriter/ELF/Writer.cpp ------------------------------------===//
+//
+//                             The LLVM Linker
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+#include "ExecutableWriter.h"
+
+using namespace llvm;
+using namespace llvm::object;
+
+namespace lld {
+
+std::unique_ptr<Writer> createWriterELF(const ELFTargetInfo &TI) {
+  using llvm::object::ELFType;
+  // Set the default layout to be the static executable layout
+  // We would set the layout to a dynamic executable layout
+  // if we came across any shared libraries in the process
+
+  if (!TI.is64Bits() && TI.isLittleEndian())
+    return std::unique_ptr<Writer>(new
+        elf::ExecutableWriter<ELFType<support::little, 4, false>>(TI));
+  else if (TI.is64Bits() && TI.isLittleEndian())
+    return std::unique_ptr<Writer>(new
+        elf::ExecutableWriter<ELFType<support::little, 8, true>>(TI));
+  else if (!TI.is64Bits() && !TI.isLittleEndian())
+    return std::unique_ptr<Writer>(new
+        elf::ExecutableWriter<ELFType<support::big, 4, false>>(TI));
+  else if (TI.is64Bits() && !TI.isLittleEndian())
+    return std::unique_ptr<Writer>(new
+        elf::ExecutableWriter<ELFType<support::big, 8, true>>(TI));
+
+  llvm_unreachable("Invalid Options!");
+}
+
+}