12754fe60SDimitry Andric //===- ELFObjectFile.cpp - ELF object file implementation -------*- C++ -*-===//
22754fe60SDimitry Andric //
32754fe60SDimitry Andric //                     The LLVM Compiler Infrastructure
42754fe60SDimitry Andric //
52754fe60SDimitry Andric // This file is distributed under the University of Illinois Open Source
62754fe60SDimitry Andric // License. See LICENSE.TXT for details.
72754fe60SDimitry Andric //
82754fe60SDimitry Andric //===----------------------------------------------------------------------===//
92754fe60SDimitry Andric //
10dff0c46cSDimitry Andric // Part of the ELFObjectFile class implementation.
112754fe60SDimitry Andric //
122754fe60SDimitry Andric //===----------------------------------------------------------------------===//
132754fe60SDimitry Andric 
14f785676fSDimitry Andric #include "llvm/Object/ELFObjectFile.h"
15139f7f9bSDimitry Andric #include "llvm/Support/MathExtras.h"
162754fe60SDimitry Andric 
17dff0c46cSDimitry Andric namespace llvm {
182754fe60SDimitry Andric using namespace object;
192754fe60SDimitry Andric 
2091bc56edSDimitry Andric ErrorOr<ObjectFile *>
2191bc56edSDimitry Andric ObjectFile::createELFObjectFile(std::unique_ptr<MemoryBuffer> &Obj) {
2291bc56edSDimitry Andric   std::pair<unsigned char, unsigned char> Ident =
2391bc56edSDimitry Andric       getElfArchType(Obj->getBuffer());
24139f7f9bSDimitry Andric   std::size_t MaxAlignment =
2591bc56edSDimitry Andric     1ULL << countTrailingZeros(uintptr_t(Obj->getBufferStart()));
26139f7f9bSDimitry Andric 
2791bc56edSDimitry Andric   std::error_code EC;
2891bc56edSDimitry Andric   std::unique_ptr<ObjectFile> R;
292754fe60SDimitry Andric   if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2LSB)
30139f7f9bSDimitry Andric #if !LLVM_IS_UNALIGNED_ACCESS_FAST
31139f7f9bSDimitry Andric     if (MaxAlignment >= 4)
3291bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, 4, false>>(
3391bc56edSDimitry Andric           std::move(Obj), EC));
34139f7f9bSDimitry Andric     else
35139f7f9bSDimitry Andric #endif
36139f7f9bSDimitry Andric     if (MaxAlignment >= 2)
3791bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, 2, false>>(
3891bc56edSDimitry Andric           std::move(Obj), EC));
39139f7f9bSDimitry Andric     else
4091bc56edSDimitry Andric       return object_error::parse_failed;
412754fe60SDimitry Andric   else if (Ident.first == ELF::ELFCLASS32 && Ident.second == ELF::ELFDATA2MSB)
42139f7f9bSDimitry Andric #if !LLVM_IS_UNALIGNED_ACCESS_FAST
43139f7f9bSDimitry Andric     if (MaxAlignment >= 4)
4491bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, 4, false>>(std::move(Obj),
4591bc56edSDimitry Andric                                                                  EC));
46139f7f9bSDimitry Andric     else
47139f7f9bSDimitry Andric #endif
48139f7f9bSDimitry Andric     if (MaxAlignment >= 2)
4991bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, 2, false>>(std::move(Obj),
5091bc56edSDimitry Andric                                                                  EC));
51139f7f9bSDimitry Andric     else
5291bc56edSDimitry Andric       return object_error::parse_failed;
532754fe60SDimitry Andric   else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2MSB)
54139f7f9bSDimitry Andric #if !LLVM_IS_UNALIGNED_ACCESS_FAST
55139f7f9bSDimitry Andric     if (MaxAlignment >= 8)
5691bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, 8, true>>(std::move(Obj),
5791bc56edSDimitry Andric                                                                 EC));
58139f7f9bSDimitry Andric     else
59139f7f9bSDimitry Andric #endif
60139f7f9bSDimitry Andric     if (MaxAlignment >= 2)
6191bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, 2, true>>(std::move(Obj),
6291bc56edSDimitry Andric                                                                 EC));
63139f7f9bSDimitry Andric     else
6491bc56edSDimitry Andric       return object_error::parse_failed;
65dff0c46cSDimitry Andric   else if (Ident.first == ELF::ELFCLASS64 && Ident.second == ELF::ELFDATA2LSB) {
66139f7f9bSDimitry Andric #if !LLVM_IS_UNALIGNED_ACCESS_FAST
67139f7f9bSDimitry Andric     if (MaxAlignment >= 8)
6891bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, 8, true>>(
6991bc56edSDimitry Andric           std::move(Obj), EC));
70139f7f9bSDimitry Andric     else
71139f7f9bSDimitry Andric #endif
72139f7f9bSDimitry Andric     if (MaxAlignment >= 2)
7391bc56edSDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, 2, true>>(
7491bc56edSDimitry Andric           std::move(Obj), EC));
75139f7f9bSDimitry Andric     else
7691bc56edSDimitry Andric       return object_error::parse_failed;
77dff0c46cSDimitry Andric   }
7891bc56edSDimitry Andric   else
7991bc56edSDimitry Andric     llvm_unreachable("Buffer is not an ELF object file!");
80dff0c46cSDimitry Andric 
8191bc56edSDimitry Andric   if (EC)
8291bc56edSDimitry Andric     return EC;
8391bc56edSDimitry Andric   return R.release();
842754fe60SDimitry Andric }
852754fe60SDimitry Andric 
862754fe60SDimitry Andric } // end namespace llvm
87