72553767d5
All <Arch>ELFFileCreateFileTraits structs are the same except its file type. That means that we don't need to pass the type traits. Instead, we can only pass file types. By doing this, we can remove copy-pasted boilerplates. llvm-svn: 234047
86 lines
3.1 KiB
C++
86 lines
3.1 KiB
C++
//===- lib/ReaderWriter/ELF/CreateELF.h -----------------------------------===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// \brief This file provides a simple way to create an object templated on
|
|
/// ELFType depending on the runtime type needed.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLD_READER_WRITER_ELF_CREATE_ELF_H
|
|
#define LLD_READER_WRITER_ELF_CREATE_ELF_H
|
|
|
|
#include "lld/Core/File.h"
|
|
#include "llvm/Object/ELF.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
|
|
namespace {
|
|
using llvm::object::ELFType;
|
|
|
|
/// \func createELF
|
|
/// \brief Create an object depending on the runtime attributes and alignment
|
|
/// of an ELF file.
|
|
///
|
|
/// \param Traits
|
|
/// Traits::create must be a template function which takes an ELFType and
|
|
/// returns an ErrorOr<std::unique_ptr<lld::File>>.
|
|
///
|
|
/// \param ident pair of EI_CLASS and EI_DATA.
|
|
/// \param maxAlignment the maximum alignment of the file.
|
|
/// \param args arguments forwarded to CreateELFTraits<T>::create.
|
|
|
|
#define LLVM_CREATE_ELF_CreateELFTraits(endian, align, is64, ...) \
|
|
new FileT<ELFType<llvm::support::endian, align, is64>>(__VA_ARGS__);
|
|
|
|
#if !LLVM_IS_UNALIGNED_ACCESS_FAST
|
|
# define LLVM_CREATE_ELF_Create(normal, low, endian, is64, ...) \
|
|
if (maxAlignment >= normal) \
|
|
return LLVM_CREATE_ELF_CreateELFTraits(endian, normal, is64, __VA_ARGS__) \
|
|
else if (maxAlignment >= low) \
|
|
return LLVM_CREATE_ELF_CreateELFTraits(endian, low, is64, __VA_ARGS__) \
|
|
else \
|
|
llvm_unreachable("Invalid alignment for ELF file!");
|
|
#else
|
|
# define LLVM_CREATE_ELF_Create(normal, low, endian, is64, ...) \
|
|
if (maxAlignment >= low) \
|
|
file = LLVM_CREATE_ELF_CreateELFTraits(endian, low, is64, __VA_ARGS__) \
|
|
else \
|
|
llvm_unreachable("Invalid alignment for ELF file!");
|
|
#endif
|
|
|
|
template <template <typename ELFT> class FileT, class... Args>
|
|
llvm::ErrorOr<std::unique_ptr<lld::File>>
|
|
createELF(std::pair<unsigned char, unsigned char> ident,
|
|
std::size_t maxAlignment, Args &&... args) {
|
|
lld::File *file = nullptr;
|
|
if (ident.first == llvm::ELF::ELFCLASS32 &&
|
|
ident.second == llvm::ELF::ELFDATA2LSB) {
|
|
LLVM_CREATE_ELF_Create(4, 2, little, false, std::forward<Args>(args)...)
|
|
} else if (ident.first == llvm::ELF::ELFCLASS32 &&
|
|
ident.second == llvm::ELF::ELFDATA2MSB) {
|
|
LLVM_CREATE_ELF_Create(4, 2, big, false, std::forward<Args>(args)...)
|
|
} else if (ident.first == llvm::ELF::ELFCLASS64 &&
|
|
ident.second == llvm::ELF::ELFDATA2MSB) {
|
|
LLVM_CREATE_ELF_Create(8, 2, big, true, std::forward<Args>(args)...)
|
|
} else if (ident.first == llvm::ELF::ELFCLASS64 &&
|
|
ident.second == llvm::ELF::ELFDATA2LSB) {
|
|
LLVM_CREATE_ELF_Create(8, 2, little, true, std::forward<Args>(args)...)
|
|
}
|
|
if (!file)
|
|
llvm_unreachable("Invalid ELF type!");
|
|
return std::unique_ptr<lld::File>(file);
|
|
}
|
|
|
|
} // end anon namespace
|
|
|
|
#undef LLVM_CREATE_ELF_CreateELFTraits
|
|
#undef LLVM_CREATE_ELF_Create
|
|
|
|
#endif
|