llvm-project/lld/lib/ReaderWriter/ELF/Mips/MipsTargetHandler.h

128 lines
3.9 KiB
C++

//===- lib/ReaderWriter/ELF/Mips/MipsTargetHandler.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_MIPS_MIPS_TARGET_HANDLER_H
#define LLD_READER_WRITER_ELF_MIPS_MIPS_TARGET_HANDLER_H
#include "DefaultTargetHandler.h"
#include "MipsELFReader.h"
#include "MipsLinkingContext.h"
#include "MipsRelocationHandler.h"
#include "MipsSectionChunks.h"
#include "TargetLayout.h"
namespace lld {
namespace elf {
/// \brief TargetLayout for Mips
template <class ELFType>
class MipsTargetLayout final : public TargetLayout<ELFType> {
public:
MipsTargetLayout(const MipsLinkingContext &ctx)
: TargetLayout<ELFType>(ctx),
_gotSection(new (_alloc) MipsGOTSection<ELFType>(ctx)),
_cachedGP(false) {}
const MipsGOTSection<ELFType> &getGOTSection() const { return *_gotSection; }
AtomSection<ELFType> *
createSection(StringRef name, int32_t type,
DefinedAtom::ContentPermissions permissions,
Layout::SectionOrder order) override {
if (type == DefinedAtom::typeGOT && name == ".got")
return _gotSection;
return DefaultLayout<ELFType>::createSection(name, type, permissions,
order);
}
/// \brief GP offset relative to .got section.
uint64_t getGPOffset() const { return 0x7FF0; }
/// \brief Get the cached value of the GP atom.
AtomLayout *getGP() {
if (!_cachedGP) {
auto gpAtomIter = this->findAbsoluteAtom("_gp_disp");
_gp = *(gpAtomIter);
_cachedGP = true;
}
return _gp;
}
private:
llvm::BumpPtrAllocator _alloc;
MipsGOTSection<ELFType> *_gotSection;
AtomLayout *_gp;
bool _cachedGP;
};
/// \brief Mips Runtime file.
template <class ELFType>
class MipsRuntimeFile final : public CRuntimeFile<ELFType> {
public:
MipsRuntimeFile(const MipsLinkingContext &context)
: CRuntimeFile<ELFType>(context, "Mips runtime file") {}
};
/// \brief TargetHandler for Mips
class MipsTargetHandler final : public DefaultTargetHandler<Mips32ElELFType> {
public:
MipsTargetHandler(MipsLinkingContext &context);
MipsTargetLayout<Mips32ElELFType> &getTargetLayout() override {
return *_targetLayout;
}
std::unique_ptr<Reader> getObjReader(bool atomizeStrings) override {
return std::unique_ptr<Reader>(new MipsELFObjectReader(atomizeStrings));
}
const MipsTargetRelocationHandler &getRelocationHandler() const override {
return *_relocationHandler;
}
std::unique_ptr<Writer> getWriter() override;
void registerRelocationNames(Registry &registry) override;
private:
static const Registry::KindStrings kindStrings[];
MipsLinkingContext &_context;
std::unique_ptr<MipsRuntimeFile<Mips32ElELFType>> _runtimeFile;
std::unique_ptr<MipsTargetLayout<Mips32ElELFType>> _targetLayout;
std::unique_ptr<MipsTargetRelocationHandler> _relocationHandler;
};
class MipsDynamicSymbolTable : public DynamicSymbolTable<Mips32ElELFType> {
public:
MipsDynamicSymbolTable(const MipsLinkingContext &context,
MipsTargetLayout<Mips32ElELFType> &layout)
: DynamicSymbolTable<Mips32ElELFType>(
context, layout, ".dynsym",
DefaultLayout<Mips32ElELFType>::ORDER_DYNAMIC_SYMBOLS),
_targetLayout(layout) {}
void sortSymbols() override {
std::stable_sort(_symbolTable.begin(), _symbolTable.end(),
[this](const SymbolEntry &A, const SymbolEntry &B) {
if (A._symbol.getBinding() != STB_GLOBAL &&
B._symbol.getBinding() != STB_GLOBAL)
return A._symbol.getBinding() < B._symbol.getBinding();
return _targetLayout.getGOTSection().compare(A._atom, B._atom);
});
}
private:
MipsTargetLayout<Mips32ElELFType> &_targetLayout;
};
} // end namespace elf
} // end namespace lld
#endif