GCC Makefile定义

Question

我有一些要测试的C代码。 它使用malloc，calloc和释放代码。 我想将这些功能更改为内部调用原始功能的自定义功能。 例如：

emxArray->size = (int *)malloc((unsigned int)(sizeof(int) * numDimensions));

会成为：

emxArray->size = (int *)myMalloc((unsigned int)(sizeof(int) * numDimensions));

myMalloc在哪里：

void* myMalloc(unsigned size)
{
    if (size < 8)
    {
        //printf("*** Bumped from %d....\n", size);
        size = 8;
    }
    allocated += size;
    return malloc(size);
}

如您所见，myMalloc在内部调用malloc。 它只是做一些额外的东西。 我想用myMalloc替换整个代码中的malloc用法。 我已经通过完成所有代码并将malloc手动替换为myMalloc来成功地做到了这一点，但这远非理想。 我将仅在测试的基础上替换此代码，因此生产代码应仅包含malloc调用。 我意识到我也可以使用脚本来执行此操作，但只想在Makefile中使用define语句：

-Dmalloc=myMalloc

但这也替换了myMalloc函数中的malloc，这会导致无限递归的情况。 我尝试将myMalloc函数中的malloc调用更改为malloc_d，并向Makefile添加了第二个定义。

-Dmalloc=myMalloc -Dmalloc_d=malloc

我在想，第一个定义不会替换malloc_d（它没有），而第二个定义只会更改malloc_d（它没有）。 我有相同的递归情况。 无论如何，用Makefile定义可以做到这一点吗？ 还是多遍预编译的情况总是会搞砸呢？

更新：

好的，因此我已经开始查看已指出的LD_PRELOAD选项。 我以为我有一个可行的解决方案，但是，我仍然遇到麻烦！ 这是我做的...

1. 我将myMalloc（）和myFree（）从主文件移到了自己的文件中。 然后，我使用以下命令将其编译到共享库中：gcc -shared -o libMyMalloc.so -fPIC myMalloc.c

2. 然后，我在主文件中添加了以下“虚拟函数”：

   void * myMalloc（unsigned size）{void * ptr; 返回ptr; }

   无效的myFree（void * ptr）{}

如您所见，它们什么也不做。

3. 我在make文件中添加了以下定义：-Dmalloc = myMalloc \\ -Dfree = myFree

4. 我编译了代码，并针对创建的libMyMalloc.so库运行了该代码：

   LD_PRELOAD = / home / rad / Desktop / myMalloc / libMyMalloc.so ./testRegress

但是，我仍然无法使其与libMyMalloc.so文件中定义的myMalloc函数一起运行。

Answer 1

最简单的解决方案是不直接在代码中调用malloc ：如果选择其他名称（例如MALLOC ），则切换到自定义分配器很简单。

示例代码：

#ifndef MALLOC
#define MALLOC malloc
#endif

对于测试版本，可以使用-DMALLOC=myMalloc 。

如果出于某种原因要保留对malloc的调用，它将变得更加复杂。 然后， 在包含所有标准库头之后，您必须添加以下内容：

#ifdef USE_MY_MALLOC
#undef malloc
#define malloc(SIZE) myMalloc(SIZE)
#endif

您可以使用parens调用标准库函数，即

void* myMalloc(unsigned size)
{
    ...
    return (malloc)(size);
}

并通过-DUSE_MY_MALLOC启用它。

---

考虑到注释中给出的其他要求，想到了两种方法：

• 预处理生成的源，以文本方式替换对malloc调用
• 拦截stdlib.h包含（假设这是MATLAB从中获取其malloc声明的位置）

您自己的stdlib.h版本看起来像这样：

#ifndef MY_STDLIB_H_
#define MY_STDLIB_H_

#include "/usr/include/stdlib.h"

#undef malloc
#define malloc(SIZE) myMalloc(SIZE)

#endif

然后，您可以有条件地将放置该文件的目录添加到包含路径。 还要注意，这不是一个特别健壮的解决方案，但是无论如何它可能对您有用。

Answer 2

您可以使用指向函数的指针。 通常情况下，使其指向malloc。 在调试模式下，让它指向您的功能。

在一些h文件中：

extern void *(*myMalloc)(size_t size);

在你们其中的一个c文件中：

#ifdef DEBUG
void *(*myMalloc)(size_t size) = dMalloc;
#else
void *(*myMalloc)(size_t size) = malloc; // derived from libc
#endif

Answer 3

我找到了解决方案，并希望分享。 感谢所有为我做出正确贡献的人。

我最终创建了自定义库代码，并将其编译为共享库：

gcc -shared -o libtmalloc.so -fPIC tmalloc.c

然后，我修改了makefile以使用共享库，并在自定义函数名称（内部调用malloc（）），malloc_t以及calloc（）和free（）中全局定义了“ malloc”：

gcc -L/home/path/to/mallocTrace -Wall -o test test.c lib/coder/*.c -lm -ltmalloc \
-Dmalloc=malloc_t \
-Dcalloc=calloc_t \
-Dfree=free_t

定义更改了对我所有与共享库中的实现链接的函数调用。 因为我使用的是共享库（已编译），所以我不必担心我的makefile定义会在我的自定义函数中引起递归调用的情况。 通过这种用法，我可以从其他工具中获取任何预先生成的C代码，并通过这些简单的makefile更改并使用我的自定义malloc跟踪库来观察内存使用情况。