ELFFile.hpp

Go to the documentation of this file.

//
// Created by Pawel Reich on 12/22/24.
//
#pragma once
 
#include <stddef.h>
#include <stdint.h>
 
class ELFFile {
 
  public:
    bool is_valid() const;
    ELFFile(void * data, size_t size);
 
    ~ELFFile();
 
    int parse() const;
    void* get_entry() const;
 
    static constexpr char ELF_SIGNATURE[] = {0x7f, 'E', 'L', 'F'};
 
    static const unsigned int EI_NIDENT = 16;
    static const unsigned int EI_CLASS = 4;
    static const unsigned int EI_DATA = 5;
    static const unsigned int SHN_UNDEF = 0;
 
    enum p_flags {
      PF_X = 0x1,
      PF_W = 0x2,
      PF_R = 0x4
    };
 
    enum p_type {
      PT_NULL = 0,
      PT_LOAD = 1,
      PT_DYNAMIC = 2,
      PT_INTERP = 3,
      PT_NOTE = 4,
      PT_SHLIB = 5,
      PT_PHDR = 6
    };
 
    enum sh_type {
      SHT_NULL = 0,
      SHT_PROGBITS = 1,
      SHT_SYMTAB = 2,
      SHT_STRTAB = 3,
      SHT_RELA = 4,
      SHT_HASH = 5,
      SHT_DYNAMIC = 6,
      SHT_NOTE = 7,
      SHT_NOBITS = 8,
      SHT_REL = 9,
      SHT_SHLIB = 10,
      SHT_DYNSYM = 11
    };
 
    enum e_type {
      ET_NONE = 0,
      ET_REL = 1,
      ET_EXEC = 2,
      ET_DYN = 3,
      ET_CORE = 4
    };
 
    enum ei_class {
      ELFCLASSNONE = 0,
      ELFCLASS32 = 1,
      ELFCLASS64 = 2
    };
 
    enum ei_data {
      ELFDATANONE = 0,
      ELFDATA2LSB = 1,
      ELFDATA2MSB = 2
    };
 
    typedef uint16_t Elf32_Half;
    typedef uint32_t Elf32_Word;
    typedef int32_t Elf32_Sword;
    typedef uint32_t Elf32_Addr;
    typedef int32_t Elf32_Off;
 
    typedef struct __attribute__((__packed__)) {
      Elf32_Word p_type;
      Elf32_Off p_offset;
      Elf32_Addr p_vaddr;
      Elf32_Addr p_paddr;
      Elf32_Word p_filesz;
      Elf32_Word p_memsz;
      Elf32_Word p_flags;
      Elf32_Word p_align;
    } Elf32_Phdr;
 
    typedef struct __attribute__((__packed__)) {
      Elf32_Word sh_name;
      Elf32_Word sh_type;
      Elf32_Word sh_flags;
      Elf32_Addr sh_addr;
      Elf32_Off sh_offset;
      Elf32_Word sh_size;
      Elf32_Word sh_link;
      Elf32_Word sh_info;
      Elf32_Word sh_addralign;
      Elf32_Word sh_entsize;
    } Elf32_Shdr;
 
    typedef struct __attribute__((__packed__)) {
      unsigned char e_ident[EI_NIDENT];
      Elf32_Half e_type;
      Elf32_Half e_machine;
      Elf32_Word e_version;
      Elf32_Addr e_entry;
      Elf32_Off e_phoff;
      Elf32_Off e_shoff;
      Elf32_Word e_flags;
      Elf32_Half e_ehsize;
      Elf32_Half e_phentsize;
      Elf32_Half e_phnum;
      Elf32_Half e_shentsize;
      Elf32_Half e_shnum;
      Elf32_Half e_shstrndx;
    } Elf32_Header;
 
    typedef struct __attribute__((__packed__)) {
      Elf32_Sword d_tag;
      union {
        Elf32_Word d_val;
        Elf32_Addr d_ptr;
      } d_un;
    } Elf32_Dyn;
 
    typedef struct __attribute__((__packed__)) {
      Elf32_Word st_name;
      Elf32_Addr st_value;
      Elf32_Word st_size;
      unsigned char st_info;
      unsigned char st_other;
      Elf32_Half st_shndx;
    } Elf32_Sym;
 
    Elf32_Header *get_header() const;
 
    Elf32_Phdr *get_program_header() const;
 
    Elf32_Phdr *get_program_header(unsigned int index) const;
 
    Elf32_Shdr *get_section_header() const;
 
    Elf32_Shdr *get_section_header(unsigned int index) const;
 
    const char *get_string_table() const;
 
  private:
    void* m_data;
    size_t m_data_sz;
};