切换至堆分配栈后函数局部变量地址异常问题求助

我正尝试从线程栈切换到堆上分配的栈，实现了一个汇编函数更新rsp寄存器，但切换栈地址后调用函数时，局部变量的地址虽然在堆内存范围内，但并非预期的新栈地址。怀疑是对调用约定的理解有误，附上相关代码及输出，求帮助解释：

相关代码 #

switch_stack.S #

.p2align 4, ,15
.global switch_stack
switch_stack:
    movq    0(%rsp), %rdx /*Save ret adr in rdx*/ 
    movq    %rdi, %rsp /* rsp points to the new stack address*/
    movq    %rdx, 0(%rsp) /*Save ret adr in new stack head*/
    ret

switch_stack.h #

extern "C" void
switch_stack(uint64_t pNewStackAddress);

main.cc #

#include <stdio.h>
#include <stdlib.h>
#include <cstdint>
#include <switch_stack.h>

void
func()
{
    uint64_t funcLocalVar = 3;
    printf("func() local var address: %p\n", &funcLocalVar);
}

int32_t
main()
{
    uint64_t mainLocalVar = 2;
    printf("main() local var address: %p\n", &mainLocalVar);

    uint64_t stackSizeByte = 8192;
    void* pNewStack = malloc(stackSizeByte);
    void* pNewStackHighAddress =
      (static_cast<char*>(pNewStack) + stackSizeByte - 1);
    printf("Stack bottom address is: %p\n", pNewStack);
    printf("Stack upper address is: %p\n", pNewStackHighAddress);

    switch_stack(reinterpret_cast<uint64_t>(pNewStackHighAddress));

    func();
    //rest of the code...

    return 0;
}

编译及输出（Windows机器） #

PS $> g++ -std=c++20 main.cc switch_stack.S -o main.exe
PS $> ./main.exe
main() local var address: 00000067553FFA80
 Stack bottom address is: 00000239B2D8D170
  Stack upper address is: 00000239B2D8F16F
func() local var address: 00000239B2D834B8

问题原因与修正方案 #

核心问题分析 #

1. 栈对齐违规：x86-64架构（包括Windows x64）要求调用函数时rsp必须保持16字节对齐。你的汇编代码直接将rsp指向堆栈的最高地址，存入返回地址后未做对齐调整，后续调用func()时栈布局不符合约定，导致局部变量地址偏移。
2. 栈生长方向适配错误：x86-64栈是向下生长的（从高地址往低地址扩展），你把返回地址存在堆栈的最高地址后，执行ret会让rsp自动加8（返回地址占8字节），此时rsp已经超出了堆分配的栈内存范围，后续栈操作只能往更低地址（合法栈区间）“倒着”扩展，最终导致局部变量地址偏离预期。
3. Windows x64影子空间未预留：Windows x64调用约定要求调用者为被调用函数预留32字节的影子空间，你的代码完全没考虑这一点，进一步加剧栈布局混乱。

修正后的汇编函数（适配Windows x64） #

.p2align 4, ,15
.global switch_stack
switch_stack:
    ; 保存原栈上的返回地址
    movq    (%rsp), %rdx
    ; 调整新栈地址：先减去8（存放返回地址），再对齐到16字节边界
    subq    $8, %rdi
    andq    $~0xF, %rdi
    ; 切换到调整后的新栈
    movq    %rdi, %rsp
    ; 将返回地址存入新栈
    movq    %rdx, (%rsp)
    ret

主函数地址计算调整 #

堆分配的栈区间是[pNewStack, pNewStack+stackSizeByte)，栈向下生长，所以直接传入堆内存的高地址即可，让汇编函数完成对齐和空间预留：

// 替换原pNewStackHighAddress的计算逻辑
void* pNewStackHighAddress = static_cast<char*>(pNewStack) + stackSizeByte;

验证说明 #

调整后，func()的局部变量地址会落在pNewStack到pNewStackHighAddress的合法区间内，且符合栈向下生长的规律（地址低于初始rsp）。

额外建议 #

• 堆分配栈建议用VirtualAlloc替代malloc，可以设置栈的读写权限并启用溢出检测，避免越界访问破坏其他堆内存。
• 若要在栈上执行复杂逻辑，还需考虑保存/恢复非易失性寄存器（如rbp、rbx等），避免寄存器状态混乱。

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