Linux内核空间内存申请函数kmalloc、kzalloc、vmalloc的区别

Table of Contents

kmalloc()

kzalloc()

vmalloc()

总结

内核中的内存申请:kmalloc、vmalloc、kzalloc、kcalloc、get_free_pages

一、kmalloc

二、get_free_pages

三、vmalloc

四、比较

五、其它函数

六、GFP标记



https://mp.weixin.qq.com/s/RXzhu1HUta7aMtd_FzEvcA

我们都知道在用户空间动态申请内存用的函数是 malloc(),这个函数在各种操作系统上的使用是一致的,对应的用户空间内存释放函数是 free()。注意:动态申请的内存使用完后必须要释放,否则会造成内存泄漏,如果内存泄漏发生在内核空间,则会造成系统崩溃。 
  那么,在内核空间中如何申请内存呢?一般我们会用到 kmalloc()、kzalloc()、vmalloc() 等,下面我们介绍一下这些函数的使用以及它们之间的区别。

kmalloc()

函数原型:

void *kmalloc(size_t size, gfp_t flags);

  kmalloc() 申请的内存位于物理内存映射区域,而且在物理上也是连续的,它们与真实的物理地址只有一个固定的偏移,因为存在较简单的转换关系,所以对申请的内存大小有限制,不能超过128KB。 
   
较常用的 flags(分配内存的方法):

  • GFP_ATOMIC —— 分配内存的过程是一个原子过程,分配内存的过程不会被(高优先级进程或中断)打断;

  • GFP_KERNEL —— 正常分配内存;

  • GFP_DMA —— 给 DMA 控制器分配内存,需要使用该标志(DMA要求分配虚拟地址和物理地址连续)。

flags 的参考用法: 
 |– 进程上下文,可以睡眠     GFP_KERNEL 
 |– 进程上下文,不可以睡眠    GFP_ATOMIC 
 |  |– 中断处理程序       GFP_ATOMIC 
 |  |– 软中断          GFP_ATOMIC 
 |  |– Tasklet         GFP_ATOMIC 
 |– 用于DMA的内存,可以睡眠   GFP_DMA | GFP_KERNEL 
 |– 用于DMA的内存,不可以睡眠  GFP_DMA |GFP_ATOMIC 
   
对应的内存释放函数为:

void kfree(const void *objp);

kzalloc()

  kzalloc() 函数与 kmalloc() 非常相似,参数及返回值是一样的,可以说是前者是后者的一个变种,因为 kzalloc() 实际上只是额外附加了 __GFP_ZERO 标志。所以它除了申请内核内存外,还会对申请到的内存内容清零。

/** * kzalloc - allocate memory. The memory is set to zero. * @size: how many bytes of memory are required. * @flags: the type of memory to allocate (see kmalloc). */static inline void *kzalloc(size_t size, gfp_t flags){    return kmalloc(size, flags | __GFP_ZERO);}

kzalloc() 对应的内存释放函数也是 kfree()。

vmalloc()

函数原型:

void *vmalloc(unsigned long size);

  vmalloc() 函数则会在虚拟内存空间给出一块连续的内存区,但这片连续的虚拟内存在物理内存中并不一定连续。由于 vmalloc() 没有保证申请到的是连续的物理内存,因此对申请的内存大小没有限制,如果需要申请较大的内存空间就需要用此函数了。

对应的内存释放函数为:

void vfree(const void *addr);

注意:vmalloc() 和 vfree() 可以睡眠,因此不能从中断上下文调用。 
  

总结

kmalloc()、kzalloc()、vmalloc() 的共同特点是:

  1. 用于申请内核空间的内存;

  2. 内存以字节为单位进行分配;

  3. 所分配的内存虚拟地址上连续;

kmalloc()、kzalloc()、vmalloc() 的区别是:

  1. kzalloc 是强制清零的 kmalloc 操作;(以下描述不区分 kmalloc 和 kzalloc)

  2. kmalloc 分配的内存大小有限制(128KB),而 vmalloc 没有限制;

  3. kmalloc 可以保证分配的内存物理地址是连续的,但是 vmalloc 不能保证;

  4. kmalloc 分配内存的过程可以是原子过程(使用 GFP_ATOMIC),而 vmalloc 分配内存时则可能产生阻塞;

  5. kmalloc 分配内存的开销小,因此 kmalloc 比 vmalloc 要快;

一般情况下,内存只有在要被 DMA 访问的时候才需要物理上连续,但为了性能上的考虑,内核中一般使用 kmalloc(),而只有在需要获得大块内存时才使用 vmalloc()。例如,当模块被动态加载到内核当中时,就把模块装载到由 vmalloc() 分配的内存上。


内核中的内存申请:kmalloc、vmalloc、kzalloc、kcalloc、get_free_pages

https://mp.weixin.qq.com/s/9vt9X9y4F8u9diiOrq-0Rg

在内核模块中申请分配内存需要使用内核中的专用API:kmalloc、vmalloc、kzalloc、kcalloc、get_free_pages;当然,设备驱动程序也不例外;

对于提供了MMU功能的处理器而言,Linux提供了复杂的内存管理系统,使得进程所能访问到的地址空间可以达到4GB;而这4GB的空间又被划分为两个部分:0GB~3GB(PAGE_OFFSET,x86中的值是0xC0000000)的区域被用作进程的用户空间,3GB~4GB的区域被用作内核空间;

在内核空间中,从3GB到vmalloc_start之间的这段地址区域作为物理内存映射区使用,该段映射区域内包含了内核镜像、物理页框表mem_map等等,比如,我们使用的系统物理内存为160MB,那么,3GB~3GB+vmalloc_start之间的区域就应该是映射的物理内存;在物理内存映射区域之后,就是虚拟内存vmalloc区域;对于160MB的系统而言,vmalloc_start的位置就应该在3GB+160MB位置附近(在物理内存映射区与vmalloc_start位置之间还存在一个8M的gap来防止越界),vmalloc_end的位置接近4GB的位置(系统会在最后的位置处保留一片128KB大小的区域专用于页面映射);

一、kmalloc

#include <linux/slab.h>
static inline void *kmalloc(size_t size, gfp_t flags);

参数:size:指定要分配的块的大小,单位是字节;flags:指定分配内存时的控制方式;
该函数用于在内核空间中分配内存使用,它的返回速度快(除非被阻塞),并且对其分配的内存不进行任何初始化(清零)操作,分配的内存区域仍然保留有他原有的内容;

kmalloc申请得到的是物理内存,位于物理内存映射区,而且在物理地址上是连续的;但是kmalloc返回的内存地址却是虚拟地址(线性地址),返回的这个虚拟地址(线性地址)与真实的物理地址之间仅仅相差一个固定的偏移值;因此,kmalloc申请得到的物理内存块的首地址与其返回的虚拟地址之间存在着比较简单的转换关系;通过内核提供的函数virt_to_phys()可以实现该虚拟地址到真实的内核物理地址之间的转换:

#define __pa(x) ((unsigned long)(x)-PAGE_OFFSET)
static inline unsigned long virt_to_phys(volatile void* address)
{
 return __pa(address);
}

参数address是kmalloc返回的一个虚拟地址;该转换过程就是虚拟地址减去3GB(PAGE_OFFSET=0xC0000000);

一般情况下,PAGE_OFFSET=3*1024*1024*1024=0xC0000000(3G);

与之对应的函数就是phys_to_virt()用于把内核物理地址转换为虚拟地址:

#define __va(x) ((void *)((unsigned long)(x)+PAGE_OFFSET))
static inline void * phys_to_virt(unsigned long address)
{
 return __va(address);
}

这两个函数都定义在include/asm-i386/io.h中;

kmalloc()函数用于小块内存的申请,最小可以申请的内存是32字节或64字节,最大可以申请的内存是128KB-16,其中,被减掉的16个字节用于存储页描述符结构;这些都依赖于体系架构所使用的页面大小;kmalloc申请的内存在物理地址上是连续的,这对于要进行DMA传输的设备来说,是非常重要的;

kmalloc()的内存分配是基于slab机制实现的,slab机制是为分配小内存而提供的一种高效的机制;但是slab机制也不是独立的,它本身也是在页分配器的基础上来划分更细粒度的内存供调用者使用;也就是说,系统先使用页分配器分配以页为最小单位的连续物理地址,然后,kmalloc()再在这个基础上根据调用者的需要进行切分的;另外,slab机制分配的内存在物理地址和虚拟地址(线性地址/逻辑地址)上都是连续的;

对于kmalloc()申请的内存,需要使用kfree()函数来释放;
备注:kmalloc是基于slab机制实现的;

二、get_free_pages

#include <asm/pages.h>
fastcall unsigned long __get_free_pages(gfp_t gfp_mask, unsigned int order)
{
 struct page * page;
 page = alloc_pages(gfp_mask, order);
 if (!page)
  return 0;
 return (unsigned long) page_address(page);
}

参数gfp_mask用于指定申请内存时的控制方式,order用于指定申请的页数;它申请的内存位于(PAGE_OFFSET,HIGH_MEMORY)之间;

__get_free_pages()函数是页面分配器提供给调用者的最底层的内存分配函数,它申请的内存也是连续的物理内存,同样位于物理内存映射区;它是基于buddy机制实现的;在使用buddy机制实现的物理内存管理系统中,最小的分配粒度(单位)也是以页为单位的;在__get_free_pages()内部通过调用alloc_pages()来分配物理内存页;

__get_free_page()函数分配的是连续的物理内存,处理的是连续的物理地址,但是返回的也是虚拟地址(线性地址);如果想要得到正确的物理地址,也需要使用virt_to_phys()可进行转换;

对于__get_free_pages()函数申请的内存,需要使用__free_pages()函数来释放;
备注:__get_free_pages是基于buddy机制实现的;

三、vmalloc

#include <linux/vmalloc.h>
void* vmalloc(unsigned long size)
{
 return __vmalloc(size, GFP_KERNEL | __GFP_HIGHMEM, PAGE_KERNEL);
}
void* __vmalloc(unsigned long size, gfp_t gfp_mask, pgprot_t prot)
{
 return __vmalloc_node(size, gfp_mask, prot, -1);
}
void* __vmalloc_node(unsigned long size, gfp_t gfp_mask, pgprot_t prot, int node)
{
 struct vm_struct *area;
 size = PAGE_ALIGN(size);
 if(!size || (size >> PAGE_SHIFT) > num_physpages)
   return NULL;
 area = get_vm_area_node(size, VM_ALLOC, node);
 if(!area)
   return NULL;
 return __vmalloc_area_node(area, gfp_mask, prot, node);
}
void* __vmalloc_area_node(struct vm_struct* area, gfp_t gfp_mask, pgprot_t prot, int node);
void* __vmalloc_area(struct vm_struct* area, gfp_t gfp_mask, pgprot_t prot)
{
 return __vmalloc_area_node(area, gfp_mask, prot, -1);
}


vmalloc()函数也是用于申请内存的,但是它申请的内存是位于vmalloc_start到vmalloc_end之间的虚拟内存;它申请的内存在虚拟地址(线性地址/逻辑地址)上是连续的,但是并不要求在物理地址上连续,并且返回的地址与物理地址之间没有简单的转换关系;

vmalloc()函数适用于大块内存的申请环境中;但是它申请的内存不能直接用于DMA传输;因为DMA传输需要使用物理地址连续的内存块;

对于vmalloc()申请的内存,需要使用vfree()函数来释放;

备注:vmalloc是基于slab机制实现的;

四、比较

1).kmalloc/__get_free_pages申请的内存块都在物理内存映射区,即在(PAGE_OFFSET,HIGH_MEMORY)之间,处理的都是物理地址,且保证在物理地址空间上是连续的;二者返回的都是虚拟地址,如果需要得到正确的物理地址,需要使用virt_to_phys()进行转换;但是,kmalloc和vmalloc都是以字节为单位进行申请,而__get_free_pages()则是以页为单位进行申请;

2).vmalloc函数申请的内存块位于虚拟内存映射区,即在(VMALLOC_START,VMALLOC_END)之间,处理的都是虚拟内存,且保证在虚拟地址空间上是连续的,但是在物理地址空间上不要求连续;一般作为交换区、模块的内存使用;

3).kmalloc和vmalloc都是基于slab机制实现的,但是kmalloc的速度比vmalloc的速度快;__get_free_pages是基于buddy机制实现的,速度也较快;

4).kmalloc用于小块内存的申请,通常,一次所能申请的内存块的大小在(32/64字节,128KB-16)之间;而vmalloc可以用于分配大块内存的场合;

5).kmalloc申请的内存块在物理地址空间上是连续的,所以它申请的内存块可以直接用于DMA传输;vmalloc申请的内存块在虚拟地址空间上连续,但是在物理地址空间上不要求连续,所以它申请的内存块不能直接用于DMA传输;

6).kmalloc申请的内存块用kfree释放;vmalloc申请的内存块用vfree释放;__get_free_pages申请的内存页用__free_pages释放;

7).kmalloc申请得到的地址称为内核逻辑地址,vmalloc申请得到的地址称为内核虚拟地址;

五、其它函数

1).static inline void *kzalloc(size_t size, gfp_t flags);
 该函数比kmalloc多了一个功能,就是会把申请得到的内存块初始化为0;

2).

static inline void* kcalloc(size_t n, size_t size, gfp_t flags)
   {
     if(n != 0 && size > ULONG_MAX / n)
        return NULL;
     return kzalloc(n * size, flags);
   }


   该函数用于申请一个数组的内存空间,并把申请得到的内存都初始化为0;

六、GFP标记

kmalloc、kzalloc、kcalloc、vmalloc、get_free_pages函数在调用时都有一个gfp_t类型的控制标记flags;这个标记用于控制申请内存时的内存分配控制方式; #include <linux/gfp.h>

GFP的标记有两种:带双下划线前缀的和不带双下划线前缀的;

不带双下划线前缀的GFP标志:

  1. GFP_ATOMIC:用于在中断上下文和进程上下文之外的其它代码中分配内存;从不睡眠;
  2. GFP_KERNEL:内核正常分配内存;可能睡眠;
  3. GFP_USER  :用于为用户空间页分配内存;可能睡眠;
  4. GFP_HIGHUSER:如同GFP_USER,但它是从高端内存中申请;
  5. GFP_NOIO和GFP_NOFS:功能如同GFP_KERNEL,但是它俩增加限制到内核能做的来满足请求;GFP_NOFS分配不允许进行任何文件系统调用,而GFP_NOIO分配根本不允许进行任何IO初始化;它俩主要用于文件系统和虚拟内存代码,那里允许一个分配睡眠,但是递归的文件系统调用会是个坏主意;

带有双下划线前缀的GFP标志:

  1. __GFP_DMA:这个标志要求分配的内存在能够进行DMA的内存区;平台依赖的;
  2. __GFP_HIGHMEM:这个标志指示分配的内存可以位于高端内存区;平台依赖的;
  3. __GFP_COLD:正常地,内存分配器尽力返回"缓冲热"的页---可能在处理器缓冲中找到的页;相反,这个标志请求一个"冷"页---在一段时间内没被使用的页;它对分配页做DMA读是很有用的,此时在处理器缓冲中出现是没用的;
  4. __GFP_NOWARN:这个标志用于分配内存时阻止内核发出警告,当一个分配请求无法满足时;
  5. __GFP_HIGH:这个标志标识了一个高优先级请求,它被允许来消耗甚至被内核保留给紧急状况的最后的内存页;
  6. __GFP_REPEAT:分配器的动作;当分配器有困难满足一个分配请求时,通过重复尝试的方式来"尽力尝试",但是分配操作仍然有可能失败;
  7. __GFP_NOFAIL:分配器的动作;当分配器有困难满足一个分配请求时,这个标志告诉分配器不要失败,尽最大努力来满足分配请求;
  8. __GFP_NORETRY:分配器的动作;当分配器有困难满足一个分配请求时,这个标志告诉分配器立即放弃,不再做任何尝试;

通常,一个或多个带双下划线前缀的标记相或,即可得到对应的不带双下划线前缀的标记;

最常用的标记就是GFP_KERNEL,它的意思就是当前的这个分配代表运行在内核空间的进程而进行的;换句话说,这意味着调用函数是代表一个进程在执行一个系统调用;使用GFP_KERNEL标记,就意味着kmalloc能够使当前进程在少内存的情况下通过睡眠来等待一个内存页;因此,一个使用GFP_KERNEL的函数必须是可重入的,且不能在原子上下文中运行;当当前进程睡眠,内核采取正确的动作来定位一些空闲的内存页,或者通过刷新缓存到磁盘或者交换出去一个用户进程的内存页;
如果一个内存分配动作发生在中断处理或内核定时器的上下文中时,当前进程就不能被设置为睡眠,也就不能再使用GFP_KERNEL标志了,此时应该使用GFP_ATOMIC标志来代替;正常地,内核试图保持一些空闲页以便来满足原子的分配;当使用GFP_ATOMIC标志时,kmalloc标志能够使用甚至最后一个空闲页;如果这最后一个空闲页不存在,那分配就会失败;

### Linux Kernel 中内存分配函数 `kmalloc` `vmalloc` 的详解及其用法 #### 1. **`kmalloc` 函数** `kmalloc` 是一种高效的内核内存分配函数,适用于分配较小的连续物理内存块。它支持原子环境下的内存分配,并且分配速度较快。 - **特点**: - 分配的是连续的物理内存。 - 支持原子级别的内存分配(通过指定标志位 `GFP_ATOMIC`)。 - 开销低,适合小规模内存分配场景。 - **语法**: ```c void *kmalloc(size_t size, gfp_t flags); ``` 参数解释: - `size`: 请求分配的字节数。 - `flags`: 内存分配标志,控制分配行为(如是否允许睡眠、是否清零等)。 - **示例代码**: ```c #include <linux/slab.h> static char *buffer; int example_function(void) { buffer = kmalloc(1024, GFP_KERNEL); // 分配 1KB 大小的内存 if (!buffer) { pr_err("Memory allocation failed!\n"); return -ENOMEM; } memset(buffer, 0, 1024); // 初始化缓冲区 pr_info("Allocated %zu bytes using kmalloc.\n", 1024); return 0; } void cleanup_example(void) { kfree(buffer); // 释放由 kmalloc 分配的内存 pr_info("Freed memory allocated with kmalloc.\n"); } ``` #### 特殊形式:`kzalloc` `kzalloc` 是 `kmalloc` 的封装版本,默认会对分配的内存进行清零操作,简化了开发者的工作流程[^2]。 - **语法**: ```c void *kzalloc(size_t size, gfp_t flags); ``` - **示例代码**: ```c static char *zeroed_buffer; int example_kzalloc(void) { zeroed_buffer = kzalloc(1024, GFP_KERNEL); // 分配并自动清零 1KB 内存 if (!zeroed_buffer) { pr_err("Memory allocation failed!\n"); return -ENOMEM; } pr_info("Allocated and cleared %zu bytes using kzalloc.\n", 1024); return 0; } void cleanup_kzalloc(void) { kfree(zeroed_buffer); // 释放由 kzalloc 分配的内存 pr_info("Freed memory allocated with kzalloc.\n"); } ``` #### 2. **`vmalloc` 函数** `vmalloc` 提供了一种灵活的方式来分配大块内存,尤其当需要跨越多个页面时非常有用。然而,它的效率相对较低,因为需要额外的时间来设置页表。 - **特点**: - 分配的是虚拟地址上的连续内存,但对应的物理内存可能是不连续的。 - 更适合大规模内存分配需求。 - 不支持原子级分配,通常在进程上下文中使用。 - **语法**: ```c void *vmalloc(unsigned long size); ``` - **示例代码**: ```c #include <linux/vmalloc.h> static char *large_buffer; int example_vmalloc(void) { large_buffer = vmalloc(1 << 20); // 分配 1MB 大小的内存 if (!large_buffer) { pr_err("Virtual memory allocation failed!\n"); return -ENOMEM; } memset(large_buffer, 0, 1 << 20); // 初始化缓冲区 pr_info("Allocated %lu bytes using vmalloc.\n", 1 << 20); return 0; } void cleanup_vmalloc(void) { vfree(large_buffer); // 释放由 vmalloc 分配的内存 pr_info("Freed virtual memory allocated with vmalloc.\n"); } ``` #### 3. **对比总结** | 功能/特性 | `kmalloc` | `vmalloc` | |-------------------|------------------------------------|-------------------------------------| | **适用范围** | 小型连续内存 | 大型非连续内存 | | **分配时间** | 较短 | 较长(需构建页表) | | **物理连续性** | 连续 | 非连续 | | **原子支持** | 支持 (`GFP_ATOMIC`) | 不支持 | 对于驱动程序开发而言,如果需要快速分配少量连续内存,推荐使用 `kmalloc/kzalloc`;而对于较大的内存需求或无法找到足够的连续物理内存的情况,则可以选择 `vmalloc`[^1][^3]。 --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值