深入解析arm64函数调用约定：从寄存器分配到性能优化

1次阅读

共计 1924 个字符，预计需要花费 5 分钟才能阅读完成。

在 arm64 开发中，对函数调用约定（AAPCS）的理解不足往往会导致一些隐蔽且难以排查的问题。最常见的情况包括寄存器污染和栈不对齐。寄存器污染通常发生在函数调用过程中，当调用者未能正确保存和恢复调用者保存的寄存器时，会导致寄存器中的值被意外修改，进而引发难以追踪的 bug。栈不对齐则更为棘手，因为 arm64 架构要求栈指针（SP）必须始终保持 16 字节对齐，否则在访问某些内存地址时会导致程序崩溃。这些问题在性能敏感的代码中尤为突出，因为错误的调用约定使用可能导致不必要的内存访问和寄存器保存 / 恢复操作，从而影响整体性能。

arm64 的 AAPCS（Procedure Call Standard for the ARM 64-bit Architecture）定义了函数调用时寄存器的使用规则。与 ARM32 相比，arm64 的寄存器数量更多，且使用方式有显著差异。以下是主要寄存器的用途划分：

x0-x7：用于传递函数参数和返回值。x0-x7 依次用于前 8 个参数，多余的参数通过栈传递。x0 还用于存放返回值。
x8：间接结果寄存器，用于返回较大的结构体。
x9-x15：临时寄存器，调用者无需保存。
x16-x17：平台寄存器，用于特定用途（如动态链接）。
x18：平台寄存器，保留给平台使用。
x19-x28：被调用者保存寄存器，调用者无需保存。
x29（FP）：帧指针，用于跟踪调用栈。
x30（LR）：链接寄存器，用于保存返回地址。

与 ARM32 相比，arm64 的寄存器数量更多，且参数传递更依赖于寄存器而非栈，这在大多数情况下能显著提升性能。

在函数调用时，参数的顺序会影响是否需要使用栈来传递额外的参数。例如，考虑以下两个函数：

// 版本 1：参数顺序不佳
void func1(int a, double b, int c, double d, int e, double f, int g, double h, int i) {// 函数体}

// 版本 2：优化后的参数顺序
void func2(int a, int c, int e, int g, int i, double b, double d, double f, double h) {// 函数体}

在版本 1 中，由于浮点参数和整型参数交替出现，可能导致部分参数需要通过栈传递。而在版本 2 中，整型参数集中在前面，浮点参数集中在后面，能更有效地利用寄存器，减少栈使用。

arm64 中，浮点参数通过 NEON 寄存器（v0-v7）传递。合理使用 NEON 寄存器可以显著提升浮点运算的性能。例如：

// 使用 NEON 寄存器传递浮点参数
float add_floats(float a, float b) {return a + b;}

对应的汇编代码可能如下：

add_floats:
    fadd s0, s0, s1  // s0 和 s1 是 NEON 寄存器，用于传递浮点参数
    ret

可变参数函数中 x8 寄存器的特殊用法 ：在可变参数函数（如 printf）中，x8 寄存器用于保存参数的数量，以便正确解析栈上的参数。
混合使用 Swift/OC 时的调用约定冲突 ：Swift 和 Objective- C 的调用约定有所不同，尤其是在结构体传递和浮点处理上。混合使用时需特别注意兼容性。
栈指针 16 字节对齐的硬性要求 ：arm64 要求栈指针（SP）必须始终保持 16 字节对齐，否则会导致程序崩溃。在编写汇编代码或处理栈帧时需特别注意。

使用 GDB 可以检查函数调用时的寄存器状态和栈布局。例如：

gdb ./your_program
break func_name
run
info registers
x/10x $sp

通过内联汇编可以验证调用约定是否符合预期。例如：

void verify_calling_convention() {
    asm volatile(
        "mov x0, #42\n"  // 设置 x0 为 42
        "bl some_func\n" // 调用函数
        "cmp x0, #42\n"  // 检查 x0 是否被修改
        "bne error\n"    // 如果 x0 被修改，跳转到错误处理
        ::: "x0"         // 告诉编译器 x0 被修改
    );
}

在 JIT（即时编译）场景下，动态生成符合 AAPCS 的代码是一个挑战。JIT 编译器需要根据目标函数的签名动态分配寄存器和栈空间，并确保调用约定的一致性。例如，当生成一个调用外部函数的代码片段时，JIT 编译器需要正确设置 x0-x7 和 v0-v7 寄存器，并处理栈对齐问题。

通过深入理解 arm64 的函数调用约定，开发者可以编写出更高效、更可靠的代码，并避免因调用约定不当导致的隐蔽问题。

正文完