blob: 34a6483b45f67e0805747b3bf0a20f851b8b3932 [file] [log] [blame]
//===- ELFWriter.h --------------------------------------------------------===//
//
// The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef MCLD_ELF_WRITER_H
#define MCLD_ELF_WRITER_H
#ifdef ENABLE_UNITTEST
#include <gtest.h>
#endif
#include <llvm/Support/ELF.h>
#include <mcld/MC/MCLDOutput.h>
namespace llvm {
class MCSectionData;
}
namespace mcld
{
class MCLDInfo;
class Layout;
class GNULDBackend;
class Relocation;
class LDSection;
/** \class ELFWriter
* \brief ELFWriter provides basic functions to write ELF sections, symbols,
* and so on.
*/
class ELFWriter
{
public:
typedef uint64_t FileOffset;
protected:
ELFWriter(GNULDBackend& pBackend)
: f_Backend(pBackend) {
}
public:
virtual ~ELFWriter() { }
GNULDBackend& target()
{ return f_Backend; }
const GNULDBackend& target() const
{ return f_Backend; }
virtual void writeELF32Header(const MCLDInfo& pInfo,
const Layout& pLayout,
const GNULDBackend& pBackend,
Output& pOutput) const;
virtual void writeELF64Header(const MCLDInfo& pInfo,
const Layout& pLayout,
const GNULDBackend& pBackend,
Output& pOutput) const;
virtual uint64_t getEntryPoint(const MCLDInfo& pInfo,
const Layout& pLayout,
const GNULDBackend& pBackend,
const Output& pOutput) const;
protected:
void emitELF32SectionHeader(Output& pOutput, MCLinker& pLinker) const;
void emitELF64SectionHeader(Output& pOutput, MCLinker& pLinker) const;
void emitELF32ProgramHeader(Output& pOutput,
const GNULDBackend& pBackend) const;
void emitELF64ProgramHeader(Output& pOutput,
const GNULDBackend& pBackend) const;
// emitShStrTab - emit .shstrtab
void emitELF32ShStrTab(Output& pOutput, MCLinker& pLinker) const;
void emitELF64ShStrTab(Output& pOutput, MCLinker& pLinker) const;
void emitSectionData(const Layout& pLayout,
const LDSection& pSection,
MemoryRegion& pRegion) const;
void emitRelocation(const Layout& pLayout,
const Output& pOutput,
const LDSection& pSection,
MemoryRegion& pRegion) const;
void emitRel(const Layout& pLayout,
const Output& pOutput,
const llvm::MCSectionData& pSectionData,
MemoryRegion& pRegion) const;
void emitRela(const Layout& pLayout,
const Output& pOutput,
const llvm::MCSectionData& pSectionData,
MemoryRegion& pRegion) const;
private:
// getSectEntrySize - compute ElfXX_Shdr::sh_entsize
uint64_t getELF32SectEntrySize(const LDSection& pSection) const;
// getSectEntrySize - compute ElfXX_Shdr::sh_entsize
uint64_t getELF64SectEntrySize(const LDSection& pSection) const;
// getSectEntrySize - compute ElfXX_Shdr::sh_link
uint64_t getSectLink(const LDSection& pSection, const Output& pOutput) const;
// getSectEntrySize - compute ElfXX_Shdr::sh_info
uint64_t getSectInfo(const LDSection& pSection, const Output& pOutput) const;
uint64_t getELF32LastStartOffset(const Output& pOutput) const;
uint64_t getELF64LastStartOffset(const Output& pOutput) const;
protected:
GNULDBackend& f_Backend;
};
} // namespace of mcld
#endif