mirror of
https://gitlab.com/BinaryHardening/cfgrip
synced 2026-07-26 12:41:08 +00:00
38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
#pragma once
|
|
#include "loader/binary.hpp"
|
|
#include <vector>
|
|
#include <string>
|
|
#include <map>
|
|
|
|
class PELoader : public Binary
|
|
{
|
|
public:
|
|
PELoader();
|
|
~PELoader() override;
|
|
|
|
bool load(const string& path) override;
|
|
BinaryFormat getFormat() const override { return BinaryFormat::PE; }
|
|
Arch getArch() const override { return m_arch; }
|
|
addr_t getEntryPoint() const override { return m_entry; }
|
|
addr_t getImageBase() const override { return m_image_base; }
|
|
vector<uint8_t> readBytes(addr_t vaddr, size_t size) const override;
|
|
vector<pair<addr_t, addr_t>> getExecutableRanges() const override;
|
|
vector<pair<addr_t, string>> getExportedFunctions() const override;
|
|
vector<ImportEntry> getImportedFunctions() const override;
|
|
const string& getPath() const override { return m_path; }
|
|
|
|
private:
|
|
bool parseHeaders();
|
|
bool parseImportTable();
|
|
bool parseExportTable();
|
|
|
|
string m_path;
|
|
Arch m_arch;
|
|
addr_t m_entry;
|
|
addr_t m_image_base;
|
|
vector<uint8_t> m_data;
|
|
vector<pair<addr_t, addr_t>> m_exec_ranges;
|
|
vector<pair<addr_t, string>> m_exports;
|
|
vector<ImportEntry> m_imports;
|
|
};
|