从零开始在纯C语言中Mock函数的方案是否可行？为何尝试未生效？

问题背景 #

尝试在纯C中Mock pow函数，编写了如下代码：

fakePow.c #

// fakePow.c
#include <math.h>
double pow(double b, double p)
{ 
    return 6.0;
}

main.c #

//main.c
#include <stdio.h>
#include <math.h>
int twice(int x)
{
    if (x <= 0)
        return 0;

    return x * 2;
}
int main()
{
    int x = pow(10, 2);
    int res = twice(x);
    printf("%d\n", res);
    return 0;
}

尝试两种编译方式均未得到预期的12，而是返回200：

1. 普通编译：

gcc -c main.c
gcc -c fakePow.c
gcc main.o fakePow.o -o main
./main

200

2. LD_PRELOAD加载共享库：

gcc -c main.c
gcc -shared -o ./fakePow.so ./fakePow.c
LD_PRELOAD=./fakePow.so ./main

200

失效原因 #

你的Mock思路本身没问题，但失效的核心是GCC对pow做了编译期优化：

• 当调用pow(10,2)时，编译器识别出这是常量表达式，直接在编译阶段计算出结果100，根本没生成调用pow函数的指令，你的Mock版本完全没被执行。
• 另外，系统的pow可能是内置函数或静态链接的优先实现，就算你手动链接自己的fakePow.o，编译器也会忽略你的版本，用内置优化后的结果。

解决办法 #

1. 禁用内置函数优化 #

编译main.c时添加-fno-builtin-pow参数，强制编译器生成真正的函数调用，而不是直接计算常量：

# 普通编译模式
gcc -c -fno-builtin-pow main.c
gcc -c fakePow.c
gcc main.o fakePow.o -o main
./main  # 输出12

# LD_PRELOAD模式
gcc -c -fno-builtin-pow main.c
gcc main.o -o main -lm  # 链接数学库
gcc -shared -o ./fakePow.so ./fakePow.c
LD_PRELOAD=./fakePow.so ./main  # 输出12

2. 避免使用常量参数 #

如果pow的参数不是编译期常量，编译器无法提前计算结果，自然会调用函数：

// 修改main.c的main函数
int main()
{
    double base = 10.0;
    double exp = 2.0;
    int x = pow(base, exp);
    int res = twice(x);
    printf("%d\n", res);
    return 0;
}

此时不管用普通编译还是LD_PRELOAD，都会输出12。

3. 弱符号标记（辅助手段） #

在fakePow.c中给系统pow标记为弱符号，让你的实现优先被选用：

// fakePow.c
#include <math.h>
__attribute__((weak)) double pow(double b, double p);
double pow(double b, double p)
{ 
    return 6.0;
}

注意：这个方法仍需配合禁用内置函数优化才能生效，否则编译器还是会直接计算常量结果。

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