zzy
741171dbba
- 在AST定义中移除函数调用结构体中的冗余name字段 - 实现完整的函数声明和定义处理流程，支持符号表查找 - 添加函数参数引用节点类型，支持参数传递和访问 - 实现函数调用的IR生成，包括参数处理和符号解析 - 添加断言确保节点有效性，提升代码健壮性 fix(ast2ir): 优化类型转换处理逻辑 - 移除多余的注释说明 - 简化参数为空检查逻辑，提高代码简洁性 - 修复函数调用时的符号表查找机制 refactor(ir): 改进IR构建器接口设计 - 修改函数构建相关API，使接口更加清晰 - 添加函数声明集合管理 - 重构内置类型缓存机制 feat(ir2mcode): 完善AMD64代码生成 - 实现函数参数到寄存器的映射 - 添加函数调用约定支持（最多4个参数） - 实现函数符号和重定位处理 - 添加栈帧管理机制 - 修正栈偏移计算 chore(ir): 清理和优化IR dump输出 - 更新节点类型描述信息 - 改进函数声明和定义的输出格式 - 修正格式化输出中的符号显示问题 style: 代码格式化和命名规范化 - 统一重定位类型枚举命名 - 优化函数参数验证和错误处理
72 lines
3.2 KiB
C
72 lines
3.2 KiB
C
#include <sccf.h>
|
||
#include <sccf2pe.h>
|
||
#include <sccf_builder.h>
|
||
#include <stdio.h>
|
||
|
||
int main() {
|
||
char data[] = "Hello, World from SCC PE Builder!\n\0";
|
||
|
||
/* clang-format off */
|
||
char code[] = {
|
||
// sub rsp, 0x28 ; 为函数调用分配栈空间
|
||
0x48, 0x83, 0xEC, 0x28,
|
||
// lea rcx, [rip + data_offset] ; 将字符串地址加载到RCX（第一个参数）
|
||
0x48, 0x8D, 0x0D, 0x00, 0x00, 0x00, 0x00,
|
||
// call qword ptr [rip + puts_iat] ; 通过IAT调用puts
|
||
0xFF, 0x15, 0x00, 0x00, 0x00, 0x00,
|
||
// add rsp, 0x28 ; 恢复栈空间
|
||
0x48, 0x83, 0xC4, 0x28,
|
||
// xor eax, eax ; 设置返回值为0
|
||
0x33, 0xC0,
|
||
// ret ; 返回
|
||
0xC3,
|
||
};
|
||
/* clang-format on */
|
||
|
||
sccf_builder_t builder;
|
||
sccf_builder_init(&builder);
|
||
sccf_sect_data_t text_section = {
|
||
.data = (u8 *)code, .size = sizeof(code), .cap = sizeof(code)};
|
||
sccf_sect_data_t data_section = {
|
||
.data = (u8 *)data, .size = sizeof(data), .cap = sizeof(data)};
|
||
sccf_builder_add_text_section(&builder, &text_section);
|
||
sccf_builder_add_data_section(&builder, &data_section);
|
||
usize str_idx =
|
||
sccf_builder_add_symbol(&builder, "str_data",
|
||
&(sccf_sym_t){
|
||
.sccf_sect_offset = 0,
|
||
.sccf_sect_type = SCCF_SECT_DATA,
|
||
.sccf_sym_bind = SCCF_SYM_BIND_GLOBAL,
|
||
.sccf_sym_size = sizeof(data),
|
||
.sccf_sym_type = SCCF_SYM_TYPE_DATA,
|
||
.sccf_sym_vis = SCCF_SYM_VIS_DEFAULT,
|
||
});
|
||
usize puts_idx =
|
||
sccf_builder_add_symbol(&builder, "puts",
|
||
&(sccf_sym_t){
|
||
.sccf_sect_offset = 0,
|
||
.sccf_sect_type = SCCF_SECT_NONE,
|
||
.sccf_sym_bind = SCCF_SYM_BIND_GLOBAL,
|
||
.sccf_sym_size = 8,
|
||
.sccf_sym_type = SCCF_SYM_TYPE_EXTERN,
|
||
.sccf_sym_vis = SCCF_SYM_VIS_DEFAULT,
|
||
});
|
||
sccf_builder_add_reloc(&builder,
|
||
(sccf_reloc_t){.addend = 4,
|
||
.offset = 7,
|
||
.sect_type = SCCF_SECT_CODE,
|
||
.sym_idx = str_idx,
|
||
.reloc_type = SCCF_RELOC_TYPE_REL});
|
||
sccf_builder_add_reloc(&builder,
|
||
(sccf_reloc_t){.addend = 4,
|
||
.offset = 13,
|
||
.sect_type = SCCF_SECT_CODE,
|
||
.sym_idx = puts_idx,
|
||
.reloc_type = SCCF_RELOC_TYPE_REL});
|
||
const sccf_t *sccf = sccf_builder_to_sccf(&builder);
|
||
|
||
scc_pe_builder_t pe_builder;
|
||
sccf2pe(&pe_builder, sccf);
|
||
scc_pe_dump_to_file(&pe_builder, __FILE__ "/../../test.exe");
|
||
}
|