66 lines
3.1 KiB
C
66 lines
3.1 KiB
C
#ifndef PE_DLL_MANAGER_H
|
||
#define PE_DLL_MANAGER_H
|
||
|
||
/* pe_dll_manager.h — PE 真实 DLL 加载管理器（SYSTEM/lib 兼容层）。
|
||
*
|
||
* PE 兼容层默认使用内置 shim 表解析 imports。shim 未命中的函数，
|
||
* 由 DLL 管理器从 SYSTEM/lib/ 加载真实 DLL 并解析导出表。
|
||
*
|
||
* 架构决策：
|
||
* - 仅 PE32+（64 位原生）启用真实 DLL 加载
|
||
* - PE32（32 位解释器）不加载真实 DLL（x86emu32 不能调 64 位代码）
|
||
* - 不修改 pe_service 结构体 ABI（通过全局注册模式）
|
||
* - 递归依赖：pe_dll_load → pe_load_image → resolve_imports → pe_dll_resolve → pe_dll_load
|
||
* - 循环检测：DLL_LOADING 状态遇到则跳过（填 unimpl stub，安全降级）
|
||
* - 内存：每个 DLL 用 kmem_alloc_aligned 分配 SizeOfImage + 文件副本（arena 不释放）
|
||
* - 缓存上限 32 个 DLL
|
||
*/
|
||
|
||
#include <utsm/types.h>
|
||
|
||
/* DLL 文件读取回调：从 SYSTEM/lib/ 读取 DLL 到内存。
|
||
* name : DLL 名（如 "foo.dll"，原样来自 PE import 表）
|
||
* out_data : 返回内存数据指针（调用方不负责释放）
|
||
* out_size : 返回字节大小
|
||
* 返回 0 成功，负值失败。 */
|
||
typedef int (*pe_dll_reader_fn)(const char *name, u8 **out_data, u32 *out_size);
|
||
|
||
/* 全局就绪标志（1=DLL 管理器已初始化，可解析真实 DLL）。
|
||
* pe_loader.c resolve_imports 检查此标志决定是否尝试真实 DLL。 */
|
||
extern int g_pe_dll_manager_ready;
|
||
|
||
/* 初始化 DLL 管理器（注册文件读取回调）。
|
||
* 在 DKM 驱动加载完成、block provider 注册后调用。 */
|
||
void pe_dll_manager_init(pe_dll_reader_fn reader);
|
||
|
||
/* 加载 DLL（含递归解析其自身 import）。
|
||
* 已加载返回缓存基址。循环依赖检测：LOADING 状态遇到则返回 -1。
|
||
* dll_name : DLL 名（如 "foo.dll"，大小写不敏感）
|
||
* out_base : 输出加载基址
|
||
* 返回 0 成功，负值失败。 */
|
||
int pe_dll_load(const char *dll_name, u64 *out_base);
|
||
|
||
/* 按名称解析导出函数。
|
||
* 若 DLL 未加载，自动调用 pe_dll_load。
|
||
* 管理器未就绪时返回 0。
|
||
* dll_name : DLL 名（大小写不敏感）
|
||
* func_name : 函数名（大小写不敏感，PE 导出表规范）
|
||
* 返回绝对地址，0=未找到。 */
|
||
u64 pe_dll_resolve(const char *dll_name, const char *func_name);
|
||
|
||
/* 带转发深度的内部变体（转发导出链 A→B→C 递归用，外部勿调）。 */
|
||
u64 pe_dll_resolve_ex(const char *dll_name, const char *func_name, int depth);
|
||
|
||
/* 在已加载 DLL 的模块基址上解析导出（GetProcAddress 真实 DLL 路径用）。
|
||
* image_base 必须是 pe_dll_load 返回的基址；func_name 与 ordinal 二选一
|
||
* （ordinal != 0 时按序号）。返回绝对地址，0=未找到。 */
|
||
u64 pe_dll_resolve_in_base(u64 image_base, const char *func_name, u32 ordinal);
|
||
|
||
/* 延迟 DllMain 控制（PE32+ 加载期 SSE/大栈未就绪，DllMain 必须延迟）。
|
||
* pe_service_run 在 pe_load_image 前调 pe_dll_defer_dllmain(1)，
|
||
* 切换到 1MB PE 大栈后调 pe_dll_run_pending_dllmains() 统一执行。 */
|
||
void pe_dll_defer_dllmain(int enable);
|
||
void pe_dll_run_pending_dllmains(void);
|
||
|
||
#endif /* PE_DLL_MANAGER_H */
|