#ifndef __SCC_PE_IDATA_H__
#define __SCC_PE_IDATA_H__

#include "scc_pe_def.h"

typedef SCC_VEC(IMAGE_THUNK_DATA64) scc_winpe_lookup_table_vec_t;

/// @brief Hint Name Table Builder
typedef struct {
    scc_pe_buffer_t data;    ///< 具体数据
    scc_hashtable_t str_map; ///< 符号名称映射到data的idx
    u32 section_offset;      ///< 在idata中的偏移
} scc_winpe_hnt_builder_t;

typedef struct {
    u32 offset;               ///< 相对代码段的偏移地址
    i32 addend;               ///< 可选的偏移量的附加值 可选默认为0
    u8 size;                  ///< 引用的地址的大小
    const char *library_name; ///< 库名称 eg. "Kernel32.dll"
    const char *symbol_name;  ///< 导入dll的符号名称 eg. "CreateFileW"
    u16 ordinal;              ///< 符号的序号 eg. 可选
} scc_winpe_extern_t;
typedef SCC_VEC(scc_winpe_extern_t) scc_winpe_extern_vec_t;

typedef SCC_VEC(const char *) scc_winpe_name_vec_t;
typedef struct {
    const char *name;
    scc_winpe_name_vec_t symbol_names;
} scc_winpe_idata_lib_t;
typedef SCC_VEC(scc_winpe_idata_lib_t) scc_winpe_idata_lib_vec_t;
typedef struct {
    scc_pe_buffer_t buffer; ///< 导入表数据
    scc_winpe_extern_vec_t externs;
    scc_hashtable_t externs_set;

    scc_winpe_hnt_builder_t hnt_builder;
    scc_hashtable_t iat_map; ///< 符号名称映射到idata的虚拟镜像地址
    scc_winpe_idata_lib_vec_t idata_libs;
} scc_winpe_idata_builder_t;

/**
 * @brief PE格式导入表构建器初始化
 * @warning 需要提前构造导入表所需的信息,不会检查导入表信息正确性
 *
 * @param builder
 * @param idata_libs
 */
void scc_pe_idata_builder_init(scc_winpe_idata_builder_t *builder,
                               scc_winpe_idata_lib_vec_t *idata_libs);

/**
 * @brief 获取导入表占据的大小
 * @warning 必须初始化导入表构建器
 *
 * @param builder
 * @return u32
 */
u32 scc_pe_reserve_idata(scc_winpe_idata_builder_t *builder);

/**
 * @brief 构造导入表
 * @warning 必须初始化导入表构建器
 *
 * @param builder
 * @param idata_range
 */
scc_pe_buffer_t scc_pe_construct_idata(scc_winpe_idata_builder_t *builder,
                                       scc_pe_section_range *idata_range);

// RVA Relative Virtual Address
static inline u64
scc_pe_idata_get_symbol_rva(scc_winpe_idata_builder_t *builder,
                            const char *symbol_name) {
    return (u64)scc_hashtable_get(&builder->iat_map, symbol_name);
}

#endif