线程内存泄漏排查：如何标记线程内存区域

核心方案：使用prctl(PR_SET_VMA) #

PR_SET_VMA是Linux内核提供的prctl操作，专为进程虚拟内存区域（VMA）设置自定义名称，标记后能在/proc/PID/maps中直观看到对应内存区域的用途，大幅提升内存泄漏排查效率。

关键参数说明 #

• 子命令使用PR_SET_VMA_ANON_NAME，专门用于给匿名内存区域（比如线程栈）设置名称
• 调用时需传入5个参数：
  1. 固定填PR_SET_VMA
  2. 子命令PR_SET_VMA_ANON_NAME
  3. 目标VMA的起始地址
  4. 目标VMA的长度
  5. 自定义名称的字符串指针
  6. 名称字符串的长度

结合pthread_create的实现步骤 #

线程栈属于匿名内存区域，创建线程后可通过pthread_getattr_np+pthread_attr_getstack获取栈的地址和大小，再调用prctl完成标记。

代码示例 #

#include <pthread.h>
#include <sys/prctl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define THREAD_STACK_TAG "worker-thread-stack"

void* thread_routine(void* arg) {
    pthread_attr_t attr;
    void* stack_base;
    size_t stack_size;

    // 获取当前线程的栈属性
    if (pthread_getattr_np(pthread_self(), &attr) != 0) {
        perror("pthread_getattr_np failed");
        return NULL;
    }
    if (pthread_attr_getstack(&attr, &stack_base, &stack_size) != 0) {
        perror("pthread_attr_getstack failed");
        pthread_attr_destroy(&attr);
        return NULL;
    }
    pthread_attr_destroy(&attr);

    // 给线程栈VMA设置自定义标记
    if (prctl(PR_SET_VMA, PR_SET_VMA_ANON_NAME, stack_base, stack_size,
              THREAD_STACK_TAG, strlen(THREAD_STACK_TAG)) != 0) {
        perror("prctl PR_SET_VMA_ANON_NAME failed");
        return NULL;
    }

    // 线程业务逻辑示例
    while (1) {
        sleep(5);
    }

    return NULL;
}

int main() {
    pthread_t tid;
    if (pthread_create(&tid, NULL, thread_routine, NULL) != 0) {
        perror("pthread_create failed");
        exit(EXIT_FAILURE);
    }

    pthread_join(tid, NULL);
    return 0;
}

验证效果 #

编译运行后，查看对应进程的/proc/PID/maps，会出现类似如下的条目：

7f2d1a5d3000-7f2d1a5f4000 rw-p 00000000 00:00 0 [worker-thread-stack]

注意事项 #

• 该特性要求Linux内核版本≥3.17，需确认系统内核版本满足要求
• 线程栈的地址和大小必须精准，pthread_getattr_np是获取当前线程栈信息的可靠方式
• 自定义名称长度不能超过64字节（内核限制），超出部分会被自动截断
• 除线程栈外，线程私有数据（TSD）等其他线程相关匿名内存区域，只要能获取到VMA的起始地址和长度，都可以用同样方式标记

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