如何编写兼容三种jemalloc使用场景的内存统计代码？

你的方案可行性 #

你的思路完全可行，但需要注意几个关键细节：

• 自定义的空mallctl必须严格匹配jemalloc的函数签名：int mallctl(const char *name, void *oldp, size_t *oldlenp, void *newp, size_t newlenp);，符号名、参数类型和返回值都不能出错，否则LD_PRELOAD的jemalloc无法正确替换它。
• 编译场景3（静态链接jemalloc）时，要通过编译宏跳过自定义mallctl的编译，比如用#ifdef NO_CUSTOM_MALLCTL包裹自定义函数，编译时加-DNO_CUSTOM_MALLCTL -ljemalloc。
• 空函数返回值要符合jemalloc的错误约定，比如返回-1，避免程序逻辑因为返回值异常而出错。

更优实现：动态检测jemalloc存在性 #

没必要写空函数，直接用动态链接的方式在运行时获取mallctl的地址，这种方式更灵活，完全兼容三种场景：

代码示例 #

#include <stdio.h>
#include <dlfcn.h>
#include <stdlib.h>

// 定义jemalloc mallctl的函数指针类型
typedef int (*mallctl_t)(const char*, void*, size_t*, void*, size_t);

void print_jemalloc_stats() {
    mallctl_t mallctl = NULL;

    // 先尝试从当前进程符号表找mallctl（对应编译时链接jemalloc的场景）
    mallctl = (mallctl_t)dlsym(RTLD_DEFAULT, "mallctl");
    if (!mallctl) {
        // 尝试加载已通过LD_PRELOAD注入的jemalloc库
        void *jemalloc_lib = dlopen("libjemalloc.so", RTLD_LAZY | RTLD_NOLOAD);
        if (jemalloc_lib) {
            mallctl = (mallctl_t)dlsym(jemalloc_lib, "mallctl");
        }
    }

    if (mallctl) {
        // 示例：获取已分配内存总量
        size_t allocated = 0;
        size_t sz = sizeof(allocated);
        if (mallctl("stats.allocated", &allocated, &sz, NULL, 0) == 0) {
            fprintf(stderr, "[jemalloc stats] Allocated: %zu bytes\n", allocated);
        }
        // 可以按需添加更多统计项，比如stats.resident、stats.metadata等
    }
    // 无jemalloc时直接跳过，不输出任何内容
}

int main() {
    // 测试内存操作
    void *buf = malloc(4096);
    print_jemalloc_stats();
    free(buf);
    return 0;
}

编译与运行方式 #

1. 仅用glibc（场景1）

   gcc -o mem_test mem_test.c -ldl
   ./mem_test
   

   无统计输出。

2. LD_PRELOAD加载jemalloc（场景2）

   gcc -o mem_test mem_test.c -ldl
   LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libjemalloc.so ./mem_test
   

   输出jemalloc统计信息。

3. 编译时链接jemalloc（场景3）

   gcc -o mem_test mem_test.c -ldl -ljemalloc
   ./mem_test
   

   输出jemalloc统计信息。

这种方式的优势 #

• 不用维护自定义空函数，彻底避免符号冲突风险。
• 运行时动态判断，不需要用编译宏区分不同场景，代码更简洁。
• 场景1和2下完全不影响glibc的malloc实现，因为没有链接-ljemalloc，malloc还是用glibc的版本。

注意事项 #

• 编译必须加-ldl链接动态加载库，否则dlopen和dlsym无法使用。
• 不同Linux发行版的jemalloc库路径可能不同，比如部分系统是/usr/lib/libjemalloc.so或/usr/lib64/libjemalloc.so，根据实际情况调整。
• 如果需要更详细的统计，可以调用多个mallctl接口，具体可参考jemalloc的官方文档。

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