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 
2039d628a0SDimitry Andric ELFObjectFileBase::ELFObjectFileBase(unsigned int Type, MemoryBufferRef Source)
2139d628a0SDimitry Andric     : ObjectFile(Type, Source) {}
2239d628a0SDimitry Andric 
2339d628a0SDimitry Andric ErrorOr<std::unique_ptr<ObjectFile>>
2439d628a0SDimitry Andric ObjectFile::createELFObjectFile(MemoryBufferRef Obj) {
2591bc56edSDimitry Andric   std::pair<unsigned char, unsigned char> Ident =
2639d628a0SDimitry Andric       getElfArchType(Obj.getBuffer());
27139f7f9bSDimitry Andric   std::size_t MaxAlignment =
2839d628a0SDimitry Andric       1ULL << countTrailingZeros(uintptr_t(Obj.getBufferStart()));
29139f7f9bSDimitry Andric 
3097bc6c73SDimitry Andric   if (MaxAlignment < 2)
3197bc6c73SDimitry Andric     return object_error::parse_failed;
3297bc6c73SDimitry Andric 
3391bc56edSDimitry Andric   std::error_code EC;
3491bc56edSDimitry Andric   std::unique_ptr<ObjectFile> R;
3597bc6c73SDimitry Andric   if (Ident.first == ELF::ELFCLASS32) {
3697bc6c73SDimitry Andric     if (Ident.second == ELF::ELFDATA2LSB)
3797bc6c73SDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, false>>(Obj, EC));
3897bc6c73SDimitry Andric     else if (Ident.second == ELF::ELFDATA2MSB)
3997bc6c73SDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, false>>(Obj, EC));
40139f7f9bSDimitry Andric     else
4191bc56edSDimitry Andric       return object_error::parse_failed;
4297bc6c73SDimitry Andric   } else if (Ident.first == ELF::ELFCLASS64) {
4397bc6c73SDimitry Andric     if (Ident.second == ELF::ELFDATA2LSB)
4497bc6c73SDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::little, true>>(Obj, EC));
4597bc6c73SDimitry Andric     else if (Ident.second == ELF::ELFDATA2MSB)
4697bc6c73SDimitry Andric       R.reset(new ELFObjectFile<ELFType<support::big, true>>(Obj, EC));
47139f7f9bSDimitry Andric     else
4891bc56edSDimitry Andric       return object_error::parse_failed;
4997bc6c73SDimitry Andric   } else {
5091bc56edSDimitry Andric     return object_error::parse_failed;
51dff0c46cSDimitry Andric   }
52dff0c46cSDimitry Andric 
5391bc56edSDimitry Andric   if (EC)
5491bc56edSDimitry Andric     return EC;
5539d628a0SDimitry Andric   return std::move(R);
562754fe60SDimitry Andric }
572754fe60SDimitry Andric 
582754fe60SDimitry Andric } // end namespace llvm
59