Read_Elf.h

#ifndef READ_ELF_H_
#define READ_ELF_H_

#include <cstdio>
#include <cstring>
#include <string>

typedef unsigned long long ULL;

using namespace std;

typedef struct
{
    unsigned char b[8];
    operator ULL()
    {
        return *((ULL *)b);
    }
    operator char *()
    {
        return (char *)b;
    }
} int64;

typedef struct
{
    unsigned char b[4];
    operator unsigned int()
    {
        return *((unsigned int *)b);
    }
    operator char *()
    {
        return (char *)b;
    }
} int32;

typedef struct
{
    unsigned char b[2];
    operator unsigned short()
    {
        return *((unsigned short *)b);
    }
    operator char *()
    {
        return (char *)b;
    }
} int16;

typedef struct
{
    unsigned char b;
} int8;

typedef int64 Elf64_Addr;
typedef int64 Elf64_Off;
typedef int64 Elf64_Xword;
typedef int64 Elf64_Sxword;
typedef int32 Elf64_Word;
typedef int32 Elf64_Sword;
typedef int16 Elf64_Half;

#define EI_CLASS 4
#define EI_DATA 5
#define EI_VERSION 6
#define EI_OSABI 7
#define EI_ABIVERSION 8
#define EI_PAD 9
#define EI_NIDENT 16

#define SHN_UNDEF 0
#define SHN_LOPROC 0xFF00
#define SHN_HIPROC 0xFF1F
#define SHN_LOOS 0xFF20
#define SHN_HIOS 0xFF3F
#define SHN_ABS 0xFFF1
#define SHN_COMMON 0xFFF2

#define ELF64_ST_BIND(i) ((i) >> 4)
#define ELF64_ST_TYPE(i) ((i)&0xf)
#define ELF64_ST_INFO(b, t) (((b) << 4) + ((t)&0xf))

typedef struct
{
    unsigned char e_ident[EI_NIDENT]; /* ELF identification */
    Elf64_Half e_type;                /* Object file type */
    Elf64_Half e_machine;             /* Machine type */
    Elf64_Word e_version;             /* Object file version */
    Elf64_Addr e_entry;               /* Entry point address */
    Elf64_Off e_phoff;                /* Program header offset */
    Elf64_Off e_shoff;                /* Section header offset */
    Elf64_Word e_flags;               /* Processor-specific flags */
    Elf64_Half e_ehsize;              /* ELF header size */
    Elf64_Half e_phentsize;           /* Size of program header entry */
    Elf64_Half e_phnum;               /* Number of program header entries */
    Elf64_Half e_shentsize;           /* Size of section header entry */
    Elf64_Half e_shnum;               /* Number of section header entries */
    Elf64_Half e_shstrndx;            /* Section name string table index */
} Elf64_Ehdr;

typedef struct
{
    Elf64_Word sh_name;       /* Section name */
    Elf64_Word sh_type;       /* Section type */
    Elf64_Xword sh_flags;     /* Section attributes */
    Elf64_Addr sh_addr;       /* Virtual address in memory */
    Elf64_Off sh_offset;      /* Offset in file */
    Elf64_Xword sh_size;      /* Size of section */
    Elf64_Word sh_link;       /* Link to other section */
    Elf64_Word sh_info;       /* Miscellaneous information */
    Elf64_Xword sh_addralign; /* Address alignment boundary */
    Elf64_Xword sh_entsize;   /* Size of entries, if section has table */
} Elf64_Shdr;

typedef struct
{
    Elf64_Word st_name;     /* Symbol name */
    unsigned char st_info;  /* Type and Binding attributes */
    unsigned char st_other; /* Reserved */
    Elf64_Half st_shndx;    /* Section table index */
    Elf64_Addr st_value;    /* Symbol value */
    Elf64_Xword st_size;    /* Size of object (e.g., common) */
} Elf64_Sym;


typedef struct
{
    Elf64_Word p_type;    /* Type of segment */
    Elf64_Word p_flags;   /* Segment attributes */
    Elf64_Off p_offset;   /* Offset in file */
    Elf64_Addr p_vaddr;   /* Virtual address in memory */
    Elf64_Addr p_paddr;   /* Reserved */
    Elf64_Xword p_filesz; /* Size of segment in file */
    Elf64_Xword p_memsz;  /* Size of segment in memory */
    Elf64_Xword p_align;  /* Alignment of segment */
} Elf64_Phdr;

typedef struct
{
    char name[50];
    ULL addr;
    ULL size;

} GlobalSymbol;

bool open_file(string fname);
bool read_elf(string fname);
void read_elf_header();
void read_elf_shst();
void read_elf_sections();
void read_elf_strtab();
void read_elf_phdr();
void read_elf_symtable();

extern FILE *file;

extern GlobalSymbol globalSymbol[100];
extern int symNum;

extern char *shsttab;   // section name string table
extern char *strtab;    // string table
extern char *symtab;   // symbol table

extern ULL cOffset;  // offset of code segment
extern ULL cSize;   // size of code segment
extern ULL cVadr;   // virtual address of code segment

extern ULL dOffset;  // offset of data segment
extern ULL dSize;   // size of data segment
extern ULL dVadr;   // virtual address of data segment

extern ULL gp;  // global pointer
extern ULL entry;   // entry address of program
extern ULL mainAddr;    // virtual address of main function 
extern ULL mainSize;    // size of main function code
extern ULL textSize;    // size of .text section
extern ULL dataAddr;    // virtual address of .data section

#endif