f7a8da3384
This adds the LinkingContext parameter to the ELFReader. Previously the flags in that were needed in the Context was passed to the ELFReader, this made it very hard to access data structures in the LinkingContext when reading an ELF file. This change makes the ELFReader more flexible so that required parameters can be grabbed directly from the LinkingContext. Future patches make use of the changes. There is no change in functionality though. llvm-svn: 228905
103 lines
3.2 KiB
C++
103 lines
3.2 KiB
C++
//===- lib/ReaderWriter/ELF/ELFReader.h -----------------------------------===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_READER_WRITER_ELF_READER_H
|
|
#define LLD_READER_WRITER_ELF_READER_H
|
|
|
|
#include "CreateELF.h"
|
|
#include "DynamicFile.h"
|
|
#include "ELFFile.h"
|
|
#include "lld/Core/Reader.h"
|
|
|
|
namespace lld {
|
|
namespace elf {
|
|
|
|
template <typename ELFT, typename ELFTraitsT, typename ContextT>
|
|
class ELFObjectReader : public Reader {
|
|
public:
|
|
typedef llvm::object::Elf_Ehdr_Impl<ELFT> Elf_Ehdr;
|
|
|
|
ELFObjectReader(ContextT &ctx, uint64_t machine)
|
|
: _ctx(ctx), _machine(machine) {}
|
|
|
|
bool canParse(file_magic magic, StringRef,
|
|
const MemoryBuffer &buf) const override {
|
|
return (magic == llvm::sys::fs::file_magic::elf_relocatable &&
|
|
elfHeader(buf)->e_machine == _machine);
|
|
}
|
|
|
|
std::error_code
|
|
loadFile(std::unique_ptr<MemoryBuffer> mb, const class Registry &,
|
|
std::vector<std::unique_ptr<File>> &result) const override {
|
|
std::size_t maxAlignment =
|
|
1ULL << llvm::countTrailingZeros(uintptr_t(mb->getBufferStart()));
|
|
auto f =
|
|
createELF<ELFTraitsT>(llvm::object::getElfArchType(mb->getBuffer()),
|
|
maxAlignment, std::move(mb), _ctx);
|
|
if (std::error_code ec = f.getError())
|
|
return ec;
|
|
result.push_back(std::move(*f));
|
|
return std::error_code();
|
|
}
|
|
|
|
const Elf_Ehdr *elfHeader(const MemoryBuffer &buf) const {
|
|
const uint8_t *data =
|
|
reinterpret_cast<const uint8_t *>(buf.getBuffer().data());
|
|
return (reinterpret_cast<const Elf_Ehdr *>(data));
|
|
}
|
|
|
|
protected:
|
|
ContextT &_ctx;
|
|
uint64_t _machine;
|
|
};
|
|
|
|
template <typename ELFT, typename ELFTraitsT, typename ContextT>
|
|
class ELFDSOReader : public Reader {
|
|
public:
|
|
typedef llvm::object::Elf_Ehdr_Impl<ELFT> Elf_Ehdr;
|
|
|
|
ELFDSOReader(ContextT &ctx, uint64_t machine)
|
|
: _ctx(ctx), _machine(machine) {}
|
|
|
|
bool canParse(file_magic magic, StringRef,
|
|
const MemoryBuffer &buf) const override {
|
|
return (magic == llvm::sys::fs::file_magic::elf_shared_object &&
|
|
elfHeader(buf)->e_machine == _machine);
|
|
}
|
|
|
|
std::error_code
|
|
loadFile(std::unique_ptr<MemoryBuffer> mb, const class Registry &,
|
|
std::vector<std::unique_ptr<File>> &result) const override {
|
|
std::size_t maxAlignment =
|
|
1ULL << llvm::countTrailingZeros(uintptr_t(mb->getBufferStart()));
|
|
auto f =
|
|
createELF<ELFTraitsT>(llvm::object::getElfArchType(mb->getBuffer()),
|
|
maxAlignment, std::move(mb), _ctx);
|
|
if (std::error_code ec = f.getError())
|
|
return ec;
|
|
result.push_back(std::move(*f));
|
|
return std::error_code();
|
|
}
|
|
|
|
const Elf_Ehdr *elfHeader(const MemoryBuffer &buf) const {
|
|
const uint8_t *data =
|
|
reinterpret_cast<const uint8_t *>(buf.getBuffer().data());
|
|
return (reinterpret_cast<const Elf_Ehdr *>(data));
|
|
}
|
|
|
|
protected:
|
|
ContextT &_ctx;
|
|
uint64_t _machine;
|
|
};
|
|
|
|
} // namespace elf
|
|
} // namespace lld
|
|
|
|
#endif // LLD_READER_WRITER_ELF_READER_H
|