如何替换 C 标准库函数？

Question

我们如何用我们自己实现的函数替换 C 标准库函数？

例如，如何将strcpy() 替换为我自己的strcpy() 实现，并让所有调用都链接到新的实现？

Answer 1

我不确定让链接器执行您想要的操作有多难，但这里有一个解决方案，它不涉及更改任何链接器设置，而是使用预处理器宏，以便任何尝试调用 strcpy 的代码实际上调用了一个名为 my_strcpy 的函数：

mystuff.h：

#define strcpy my_strcpy
char * my_strcpy(char * dst, const char * src);

my_strcpy.c:

#include <mystuff.h>
char * my_strcpy(char * dst, const char * src);
{
    ...
}

my_code.c:

#include <mystuff.h>

int main()
{
   /* Any call to strcpy will look like a normal call to strcpy
      but will actually call my_strcpy. */
}

Answer 2

至少对于 GCC 和 glibc，标准 C 函数的符号是​​ weak，因此您可以覆盖它们。例如，

strcpy.c:

#include <string.h>
#include <stdio.h>

char * strcpy(char *dst, const char *src)
{
  char *d = dst;
  while (*src) {
    *d = *src;
    d++;
    src++;
  }
  printf("Called my strcpy()\n");

  return (dst);
}

int main(void)
{
  char foo[10];
  strcpy(foo, "hello");

  puts(foo);

  return 0;
}

然后像这样构建它：

gcc -fno-builtin -o strcpy strcpy.c

然后：

$ ./strcpy 
Called my strcpy()
hello

请注意此处-fno-builtin 的重要性。如果你不使用这个，GCC 会将strcpy() 调用替换为一个内置函数，其中 GCC 有一个数字。

我不确定这是否适用于其他编译器/平台。

Answer 3

如果您在 Linux 上，可以尝试使用 LD_PRELOAD。