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;
};