table_manifest.hpp

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <optional>
#include <string>
#include <string_view>
#include <unordered_map>
#include <vector>
 
namespace ts {

struct DllIdentity {
    std::string file_name;
    std::string product;
    std::string version;
    std::int64_t size = 0;
    std::string sha256;
    std::string sha1;
    std::string md5;
    std::uint32_t pe_timestamp = 0;
};

struct TableEntry {
    std::string name;
    std::string symbol;
    std::string subsystem;
    std::string dtype;
    std::string shape;
    std::int32_t size = 0;
    std::int64_t file_offset = 0;
    std::optional<std::int64_t> va;
    std::optional<std::int64_t> section_adjust;
    std::string match = "full";
    std::string purpose;
};

struct LiveRegion {
    std::string name;
    std::string symbol;
    std::string subsystem;
    std::string dtype;
    std::int64_t file_offset = 0;
    std::optional<std::int64_t> size;
    std::string purpose;
};

class TableManifest {
public:
    [[nodiscard]] static const TableManifest& defaults();

    [[nodiscard]] static TableManifest parse(std::string_view json);

    [[nodiscard]] const DllIdentity& dll() const noexcept { return dll_; }

    [[nodiscard]] std::int64_t image_base() const noexcept { return image_base_; }

    [[nodiscard]] const std::vector<TableEntry>& cached_tables() const noexcept { return tables_; }

    [[nodiscard]] const std::vector<LiveRegion>& live_regions() const noexcept { return regions_; }

    [[nodiscard]] const TableEntry& table(std::string_view name) const;

    [[nodiscard]] const LiveRegion& region(std::string_view name) const;
 
private:
    TableManifest() = default;
 
    void index();

    struct StringHash {
        using is_transparent = void;
 
        [[nodiscard]] std::size_t operator()(std::string_view text) const noexcept
        {
            return std::hash<std::string_view>{}(text);
        }
    };
 
    using Index = std::unordered_map<std::string, std::size_t, StringHash, std::equal_to<>>;
 
    DllIdentity dll_;
    std::int64_t image_base_ = 0;
    std::vector<TableEntry> tables_;
    std::vector<LiveRegion> regions_;
    Index by_name_;
    Index regions_by_name_;
};
 
} // namespace ts