发现内存泄漏：如何用kmemleak与kasan定位驱动中的野指针

Linux内核驱动，内存泄漏与野指针是两大顽疾。内存泄漏会导致系统资源逐渐耗尽，而野指针则可能引发不可预知的崩溃或数据损坏。本文将深入解析kmemleak与KASAN(Kernel Address Sanitizer)的工作原理，并通过C语言示例展示如何利用这两种工具定位驱动中的野指针问题。

一、内存泄漏与野指针的本质

内存泄漏指动态分配的内存未被释放，导致系统无法回收资源。野指针则指向无效内存地址，可能源于未初始化、越界访问或释放后未置空。在驱动开发中，野指针常表现为：

未初始化指针：指针变量未初始化，其值随机指向非法地址。

释放后访问：内存释放后指针未置空，继续访问导致非法操作。

越界访问：指针超出数组或结构体边界，访问未分配内存。

二、kmemleak：内存泄漏的“侦探”

原理

kmemleak是Linux内核提供的内存泄漏检测工具，通过标记“根集”(全局变量、栈、寄存器等)并周期性扫描内存，构建从根集到已分配内存块的引用图。若某内存块无法从根集到达，则判定为疑似泄漏。

关键步骤：

标记根集：将已知的合法内存区域(如全局变量)标记为起始点。

扫描内存：遍历内存(包括SLAB、vmalloc等区域)，寻找可能是指针的值。

构建引用图：若某值指向已分配内存块的起始地址，则标记该块为“可达”。

报告泄漏：不可达的内存块被标记为泄漏，并记录分配地址、大小及调用栈。

应用示例

假设驱动中存在以下内存泄漏：

static void *leaky_alloc(void) {

void *ptr = kmalloc(1024, GFP_KERNEL);

if (!ptr) return NULL;

// 忘记释放ptr

return ptr;

}

使用kmemleak定位：

编译内核：启用CONFIG_DEBUG_KMEMLEAK选项。

触发扫描：通过/sys/kernel/debug/kmemleak接口手动触发扫描。

分析报告：报告会显示泄漏内存的分配地址及调用栈，指向leaky_alloc函数。

优点：无需修改代码，对系统性能影响较小，能检测大多数真正的内存泄漏。

局限：可能误报(如特殊存储的指针未被识别)或漏报(无法检测逻辑泄漏)。

三、KASAN：野指针的“终结者”

原理

KASAN通过编译时插桩和影子内存(Shadow Memory)技术，实时检测内存越界访问和使用已释放内存的行为。其核心机制包括：

影子内存：为每8字节物理内存分配1字节影子内存，记录访问状态(如是否可寻址)。

插桩检查：在每次内存访问前插入检查代码，验证目标地址的影子内存状态。

错误报告：若检测到非法访问(如越界或use-after-free)，立即触发内核异常并打印详细错误信息。

关键技术：

通用模式(Generic KASAN)：字节级精度，但开销较大(2x-3x性能下降)。

软件标签模式(SW_TAGS)：适用于ARM64等平台，精度较低但开销小。

应用示例

假设驱动中存在野指针越界访问：

static void wild_pointer_access(void) {

int arr[10] = {0};

int *ptr = arr;

for (int i = 0; i <= 10; i++) { // 越界访问

ptr[i] = i;

}

}

使用KASAN定位：

编译内核：启用CONFIG_KASAN选项(如CONFIG_KASAN_GENERIC)。

运行测试：触发wild_pointer_access函数执行。

捕获错误：KASAN会立即报告越界访问，显示错误地址、访问类型及调用栈。

优点：实时检测，能精准定位野指针问题，极大缩短调试时间。

局限：性能开销较大，仅适用于开发和测试环境。

四、实战：结合kmemleak与KASAN定位复杂问题

假设驱动中存在以下复合问题：

内存泄漏：动态分配的内存未释放。

野指针：释放后未置空的指针被越界访问。

static void *complex_leak_and_wild(void) {

void *ptr1 = kmalloc(1024, GFP_KERNEL);

void *ptr2 = kmalloc(512, GFP_KERNEL);

if (!ptr1 || !ptr2) goto fail;

kfree(ptr1); // 释放ptr1但未置空

ptr1 = NULL; // 实际代码中可能遗漏此行

// 野指针越界访问

int *wild_ptr = (int *)ptr2;

for (int i = 0; i <= 128; i++) { // 越界访问ptr2

wild_ptr[i] = i;

}

return ptr2; // 返回ptr2但未释放，导致泄漏

fail:

if (ptr1) kfree(ptr1);

if (ptr2) kfree(ptr2);

return NULL;

}

定位步骤：

启用KASAN：编译内核时启用CONFIG_KASAN，运行测试触发越界访问错误，定位到complex_leak_and_wild函数中的野指针问题。

修复野指针：在释放ptr1后立即置空，并修正越界访问逻辑。

启用kmemleak：编译内核时启用CONFIG_DEBUG_KMEMLEAK，运行测试后扫描内存泄漏，定位到complex_leak_and_wild函数中未释放的ptr2。

修复内存泄漏：在函数返回前释放ptr2。

五、总结

kmemleak与KASAN是Linux内核驱动开发中不可或缺的调试工具：

kmemleak：擅长检测内存泄漏，通过全局扫描和引用图分析，帮助开发者定位未释放的内存块。

KASAN：专注于实时检测野指针问题，通过影子内存和插桩技术，精准捕获越界访问和使用已释放内存的行为。

结合两者使用，可覆盖驱动开发中的主要内存问题场景，显著提升代码质量与稳定性。在实际开发中，建议：

在开发阶段启用KASAN，实时捕获野指针问题。

在测试阶段启用kmemleak，定期扫描内存泄漏。

通过/sys/kernel/debug/kmemleak和内核日志分析问题，结合调用栈定位根因。