如何在C++中识别foo函数的第二次递归调用（不修改函数/静态变量）

问题描述 #

用户提供的目标代码如下：

int foo(){
    if(<THIS IS THE SECOND TIME YOU CALL FOO>) // 需要替换成判断条件
        return 1;
    else
        return foo();
}

核心需求是：编写一个判断条件，识别当前是否是foo()的第二次调用，要求禁止使用静态/全局变量，且不得修改foo()函数的其他代码，希望通过栈帧或调用者栈帧相关的信息实现。

解决方案：利用调用者返回地址的特征 #

在x86/x86_64架构下，可借助GCC内置函数__builtin_return_address(n)获取栈帧中的返回地址，通过判断调用者的返回地址是否属于foo()函数的代码段，来区分第一次与第二次调用：

• 第一次调用foo()时，调用者是外部函数（比如main），返回地址指向外部函数的代码，不在foo()的地址范围内。
• 第二次调用foo()时，调用者是foo()自身，返回地址指向foo()内部return foo();语句的下一条指令，属于foo()的代码段。

替换后的代码如下：

#include <cstdint>

int foo(){
    if (__builtin_return_address(1) >= reinterpret_cast<uintptr_t>(&foo) && 
        __builtin_return_address(1) < reinterpret_cast<uintptr_t>(&foo) + 0x100)
        return 1;
    else
        return foo();
}

说明 #

1. __builtin_return_address(1)：获取当前foo()调用者的返回地址，参数1表示向上追溯1层调用栈。
2. 代码中的0x100是预估的foo()函数二进制大小，可根据实际情况调整；如果需要更精准的函数范围，可借助链接器脚本或编译器扩展获取函数的起始与结束地址。
3. 该方案依赖GCC编译器与x86/x86_64架构，不同编译器或架构可能需要调整实现方式，但核心思路均基于栈帧的返回地址特征。

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