2025-04-08 22:51:50 -07:00

89 lines
1.8 KiB
C++

#pragma once
#include "ELF_Architecture.h"
#include "ELF_Section.h"
#include "ELF_Program.h"
#include <ehs/Serializer.h>
/// OS' ABI Types.
#define ELFH_ABI_SYSTEMV 0x00
/// ELF Binary Types.
#define ELFH_TYPE_EXEC 0x02
#define ELFH_TYPE_DYN 0x03
/// True header sizes.
#define ELFH_SIZE 64
class ELF
{
private:
ehs::UInt_8 bitDepth;
ehs::UInt_8 endianness;
ehs::UInt_8 abi;
ehs::UInt_16 type;
ehs::UInt_16 arch;
ehs::UInt_16 entryPoint;
ehs::Vector<ELF_Program, ehs::UInt_16> programs;
ehs::Vector<ELF_Section, ehs::UInt_16> sections;
public:
ELF();
ELF(const ehs::UInt_8 &bitDepth, const ehs::UInt_8 &end, const ehs::UInt_8 &abi, const ehs::UInt_16 &type, const ehs::UInt_16 &arch);
ELF(ehs::Serializer<ehs::UInt_64>& data);
ELF(ELF&& header) noexcept;
ELF(const ELF& header);
ELF& operator=(ELF&& header) noexcept;
ELF& operator=(const ELF& header);
ehs::UInt_8 GetBitDepth() const;
ehs::UInt_8 GetEndianness() const;
ehs::UInt_8 GetABI() const;
ehs::UInt_16 GetType() const;
ehs::UInt_16 GetArchitecture() const;
void SetEntryPoint(ehs::UInt_16 programIndex);
ELF_Program* GetEntryPoint() const;
void AddProgram(ELF_Program newProgram);
bool HasSection(ehs::UInt_64 hashId);
bool HasSection(ehs::Str_8& id);
void AddSection(ELF_Section newSection);
ELF_Section* GetSection(ehs::UInt_64 sectionHashId) const;
ELF_Section* GetSection(const ehs::Str_8& sectionId) const;
ehs::UInt_64 GetLastSegmentOffset() const;
ehs::Vec2_u64 SegmentRange(ehs::UInt_64 sectionHashId);
ehs::Vec2_u64 SegmentRange(const ehs::Str_8& sectionId);
ehs::Serializer<ehs::UInt_64> Serialize() const;
private:
void InitializeSections();
ehs::UInt_64 RetrieveSectionsOffset() const;
ehs::UInt_16 FindShStrTabIndex() const;
ELF_Section* FindShStrTab() const;
};