66 lines
2.3 KiB
C
66 lines
2.3 KiB
C
#ifndef PE_SHIM_H
|
||
#define PE_SHIM_H
|
||
|
||
/* pe_shim.h — Windows API shim 表。
|
||
*
|
||
* 将常见 Windows DLL 函数映射到 Deshab 等价实现：
|
||
* kernel32.dll!WriteFile → 写串口 + 帧缓冲
|
||
* kernel32.dll!HeapAlloc → kmem_alloc
|
||
* msvcrt.dll!printf → 串口输出
|
||
* ...
|
||
*
|
||
* 64位 PE：shim 用 __attribute__((ms_abi)) 声明，PE 代码以 MS x64 ABI 调用，
|
||
* IAT 直接填 shim 函数地址，原生执行。
|
||
* 32位 PE：解释器在 CALL IAT 时拦截，从模拟栈读参数调用 fn32，结果写 EAX。
|
||
*/
|
||
|
||
#include <utsm/types.h>
|
||
|
||
/* 32位 shim：从模拟栈读取参数（cdecl 语义，调用者清栈）。
|
||
* esp 指向模拟栈（第一个参数在 esp[0]）。返回值 = EAX。 */
|
||
typedef u64 (*pe_shim_fn32)(void *esp);
|
||
|
||
/* shim 表条目 */
|
||
typedef struct pe_shim_entry {
|
||
const char *dll; /* 小写 DLL 名，如 "kernel32.dll" */
|
||
const char *func; /* 函数名，如 "WriteFile" */
|
||
void *fn64; /* 64位 ms_abi 函数地址（可为 NULL） */
|
||
pe_shim_fn32 fn32; /* 32位实现（可为 NULL） */
|
||
} pe_shim_entry;
|
||
|
||
/* 按 dll+func 名查找 shim。返回表索引，-1 未找到。 */
|
||
int pe_shim_lookup(const char *dll, const char *func);
|
||
|
||
/* 获取 shim 表条目 */
|
||
const pe_shim_entry *pe_shim_get(int index);
|
||
|
||
/* 获取 shim 表大小 */
|
||
int pe_shim_count(void);
|
||
|
||
/* 生成一个唯一的"未实现 stub"地址。
|
||
* 64位：返回一个固定 stub 函数地址，被调用时日志告警并返回 0。
|
||
* 32位：返回一个特殊标记值，解释器识别后日志告警。 */
|
||
u64 pe_shim_unimpl_stub(void);
|
||
|
||
/* 初始化 shim 子系统（注册全局状态，如 cmdline） */
|
||
void pe_shim_init(const char *cmdline);
|
||
|
||
/* ExitProcess 实现：通过 __builtin_longjmp 跳回 pe_run。
|
||
* 64位和32位共用。exit_code = 进程退出码。 */
|
||
void pe_shim_exit_process(u64 exit_code);
|
||
|
||
/* pe_run 调用：保存上下文（__builtin_setjmp）。
|
||
* 返回 0 = 首次（继续执行 PE），1 = ExitProcess 触发（longjmp 回来）。 */
|
||
int pe_shim_setup_exit(void);
|
||
|
||
/* 获取上次 ExitProcess 的退出码 */
|
||
u64 pe_shim_get_exit_code(void);
|
||
|
||
/* 判断地址是否为 ExitProcess shim（解释器用） */
|
||
int pe_shim_is_exit_call(u64 target);
|
||
|
||
/* 获取 cmdline（shim 内部用） */
|
||
const char *pe_shim_get_cmdline(void);
|
||
|
||
#endif /* PE_SHIM_H */
|