zzy
6ebf0c48e3
新增PE库的基础设施包括： - 创建README.md文档，说明代码参考rust cargo的object库实现 - 配置cbuild.toml包依赖，添加scc_utils依赖项 - 定义scc_pe_builder.h头文件，包含PE构建器的数据结构和API函数 - 实现PE文件的段管理、头信息处理和文件写入功能
78 lines
2.4 KiB
C
78 lines
2.4 KiB
C
#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
|