如何强制编译器在函数返回时保持参数寄存器不变？

问题背景 #

我正在编写简易JIT的特殊代码，包含若干字节码处理函数，运行时生成的代码会直接调用这些函数。目前尝试了两种方案，但都存在问题：

• 方案一：通过PRELUDE/POSTLUDE的汇编提示引导编译器保留寄存器值，小函数中有效，但内存增长等复杂大处理函数中失效。
• 方案二：将寄存器绑定到特定位置，效果更好，但GCC文档明确指出该方式不适用于此类场景。
• 额外尝试：考虑过使用[[clang::musttail]]调用dummy函数，但会产生额外开销。

希望找到标准的实现方式。

现有实现代码 #

方案一代码 #

#define HANDLER(name)                                                         
    void name(WasmMemory *memory, WasmValue *stack, void **misc,              
              uint64_t tmp1, uint64_t tmp2)
#define PARAMS memory, stack, misc, tmp1, tmp2
#define PRELUDE                                                               
    start: asm volatile("" :: "r"(memory), "r"(stack), "r"(misc));
#define POSTLUDE                                                              
    end:                                                                      
    asm goto("" :: "r"(memory), "r"(stack), "r"(misc) ::start, end);           
    return

HANDLER(memorysize) {
    PRELUDE;
    *stack++ = memory->size();
    POSTLUDE;
}

运行时生成代码 #

call memorysize
call i32add
call i32mul
movabs rcx, 17 ;; rcx is tmp1 register, used in i32const
call i32const

非跨平台尝试方案代码 #

#define HANDLER(name)                                                         
    void name(WasmMemory *_memory, WasmValue *_stack, void **_misc,              
              uint64_t tmp1, uint64_t tmp2)
#define PARAMS memory, stack, misc, tmp1, tmp2
#define PRELUDE                                                               
    register auto memory asm("rdi") = _memory; 
    register auto stack asm("rsi") = _stack; 
    register auto misc asm("rdx") = _misc;
#define POSTLUDE                                                              
    return

标准实现方案建议 #

1. 裸函数（Naked Functions）实现 #

GCC和Clang均支持__attribute__((naked))属性，用于定义无需编译器自动生成 prologue/epilogue 的裸函数。你可以直接在函数内编写汇编，完全控制寄存器的使用和保留逻辑，完美匹配JIT生成代码的调用规则：

__attribute__((naked))
void memorysize(WasmMemory *memory, WasmValue *stack, void **misc, uint64_t tmp1, uint64_t tmp2) {
    asm volatile (
        // 手动实现逻辑：调用memory->size()，写入栈顶后移动栈指针
        "movq  (%rdi), %rax;"       // 假设memory->size()的函数指针是第一个成员
        "callq *%rax;"
        "movq  %rax, (%rsi);"
        "addq  $8, %rsi;"           // stack++
        "retq;"                     // 直接返回，JIT代码可继续使用寄存器
        ::: "rax"                   // 声明被修改的寄存器
    );
}

这种方式完全绕过编译器的寄存器分配，精准控制寄存器生命周期，无额外性能开销。

2. 带严格约束的汇编提示 #

如果不想完全手写汇编，可以使用asm volatile配合+r约束，同时通过noinline和优化属性确保编译器保留寄存器值：

__attribute__((noinline, optimize("O2")))
void memorysize(WasmMemory *memory, WasmValue *stack, void **misc, uint64_t tmp1, uint64_t tmp2) {
    // 告诉编译器这些变量必须保留在寄存器中且可能被修改
    asm volatile ("" : "+r"(memory), "+r"(stack), "+r"(misc));
    *stack++ = memory->size();
    asm volatile ("" : "+r"(memory), "+r"(stack), "+r"(misc));
}

+r约束强制编译器将变量固定在寄存器中，避免复杂函数优化时将寄存器值溢出到栈，可靠性比空汇编提示更高。

3. 适配标准调用约定 #

让JIT生成代码贴合目标平台的标准调用约定：

• x86_64平台使用System V AMD64约定：将memory、stack、misc、tmp1分别放入rdi、rsi、rdx、rcx寄存器后调用处理函数。
• 处理函数按标准方式声明参数，编译器会自动遵循约定处理寄存器传递，无需额外绑定或提示，兼容性和稳定性最优。

内容的提问来源于stack exchange，提问作者csjh