[Object][ELF] Add a maximum alignment. This is used by createELFObjectFile to create...

[oota-llvm.git] / lib / Object / ELFObjectFile.cpp

diff --git a/lib/Object/ELFObjectFile.cpp b/lib/Object/ELFObjectFile.cpp
index 663b84ec8b1f2d160b1e7074860a4305abfbd118..2c8c1b16d11c1f6aa461e136aeadb0a175c23646 100644 (file)
--- a/lib/Object/ELFObjectFile.cpp
+++ b/lib/Object/ELFObjectFile.cpp
@@ -12,6 +12,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "llvm/Object/ELF.h"
+#include "llvm/Support/MathExtras.h"
 
 namespace llvm {
 
@@ -22,16 +23,37 @@ ObjectFile *ObjectFile::createELFObjectFile(MemoryBuffer *Object) {
   std::pair<unsigned char, unsigned char> Ident = getElfArchType(Object);
   error_code ec;
 
+  std::size_t MaxAlignment =
+    1ULL << CountTrailingZeros_64(uintptr_t(Object->getBufferStart()));
+
   if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2LSB)
-    return new ELFObjectFile<support::little, false>(Object, ec);
+    if (MaxAlignment >= 4)
+      return new ELFObjectFile<support::little, 4, false>(Object, ec);
+    else if (MaxAlignment >= 2)
+      return new ELFObjectFile<support::little, 2, false>(Object, ec);
+    else
+      llvm_unreachable("Invalid alignment for ELF file!");
   else if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2MSB)
-    return new ELFObjectFile<support::big, false>(Object, ec);
+    if (MaxAlignment >= 4)
+      return new ELFObjectFile<support::big, 4, false>(Object, ec);
+    else if (MaxAlignment >= 2)
+      return new ELFObjectFile<support::big, 2, false>(Object, ec);
+    else
+      llvm_unreachable("Invalid alignment for ELF file!");
   else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2MSB)
-    return new ELFObjectFile<support::big, true>(Object, ec);
+    if (MaxAlignment >= 8)
+      return new ELFObjectFile<support::big, 8, true>(Object, ec);
+    else if (MaxAlignment >= 2)
+      return new ELFObjectFile<support::big, 2, true>(Object, ec);
+    else
+      llvm_unreachable("Invalid alignment for ELF file!");
   else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2LSB) {
-    ELFObjectFile<support::little, true> *result =
-          new ELFObjectFile<support::little, true>(Object, ec);
-    return result;
+    if (MaxAlignment >= 8)
+      return new ELFObjectFile<support::little, 8, true>(Object, ec);
+    else if (MaxAlignment >= 2)
+      return new ELFObjectFile<support::little, 2, true>(Object, ec);
+    else
+      llvm_unreachable("Invalid alignment for ELF file!");
   }
 
   report_fatal_error("Buffer is not an ELF object file!");