kmalloc介绍

在Linux内核编程中，动态内存分配是一种常见的技术，用于在运行时根据需要分配和释放内存。内核提供了多种内存分配函数，其中kmalloc()是最常用的一种。

kmalloc()

kmalloc()函数用于在内核中分配内存，其原型定义在<linux/slab.h>中。这个函数通常用于分配小块内存，其大小通常不超过几千字节。

函数原型

void *kmalloc(size_t size, gfp_t flags);

• size: 要分配的内存大小，以字节为单位。
• flags: 分配行为的标志，这些标志影响内存分配的行为。

gfp_t标志

gfp_t（Get Free Page flags）是一组位标志，用于控制内存分配的行为。常见的标志包括：

• GFP_KERNEL: 这是进行内核分配时常用的标志。如果当前没有足够的内存，该标志允许内核休眠，等待内存可用。
• GFP_ATOMIC: 用于中断上下文中，不能睡眠的情况。它不会导致调用者睡眠，适用于中断处理程序或其他不能阻塞的代码中。
• GFP_NOWAIT: 分配内存时不会睡眠，如果没有足够的内存立即返回NULL。

使用示例

struct my_struct {
    int data;
    char name[100];
};

struct my_struct *ptr = kmalloc(sizeof(struct my_struct), GFP_KERNEL);
if (!ptr) {
    // 处理分配失败的情况
} else {
    // 使用分配的内存
    ptr->data = 123;
    strcpy(ptr->name, "example");
}

内存释放

分配的内存使用完毕后，应该使用kfree()函数释放，以避免内存泄漏。

kfree()函数

释放由kmalloc()分配的内存。

kfree(ptr);

注意事项

• 使用kmalloc()时应确保正确处理内存分配失败的情况。
• 分配的内存需要在不再使用时通过kfree()释放，避免内存泄漏。
• 选择正确的gfp_t标志对于保证代码在正确的上下文中运行非常重要，例如在中断上下文中不应使用可能导致休眠的GFP_KERNEL。

动态内存分配在内核模块开发中非常重要，合理使用这些机制可以确保内核的稳定性和效率。