malloc 函数详解

不少学过C的人对malloc都不是很了解,知道使用malloc要加头文件,知道malloc是分配一块连续的内存,知道和free函数是一块儿用的。可是可是:linux

一部分人仍是将:malloc看成系统所提供的或者是C的关键字,事实上:malloc只是C标准库中提供的一个普通函数git

并且不少不少人都对malloc的具体实现机制不是很了解。
程序员



1,关于malloc以及相关的几个函数算法

      #include <stdlib.h>(Linux下)

       void *malloc(size_t size);
       void free(void *ptr);
       void *calloc(size_t nmemb, size_t size);
       void *realloc(void *ptr, size_t size);

      也能够这样认为(window下)原型:extern void *malloc(unsigned int num_bytes);
编程

                                                      头文件:#include <malloc.h>或者#include <alloc.h>二者的内容是彻底同样的。数组

       若是分配成功:则返回指向被分配内存空间的指针缓存

       否则,返回空指针NULL。数据结构

       同时,当内存再也不使用的时候,应使用free()函数将内存块释放掉。app


       关于:void *,表示未肯定类型的指针。C,C++规定,void *类型能够强转为任何其余类型的的指针。ide


malloc returns a void pointer to the allocated space, or NULL if there is insufficient memory available. To return a pointer to a type other than void, use a type cast on the return value. The storage space pointed to by the return value is guaranteed to be suitably aligned for storage of any type of object. If size is 0, malloc allocates a zero-length item in the heap and returns a valid pointer to that item. Always check the return from malloc, even if the amount of memory requested is small.


         关于void *的其余说法:

         void * p1;

         int *p2;

         p1 = p2;                 就是说其余任意类型均可以直接赋值给它,无需进行强转,可是反过来不能够。


malloc:

malloc分配的内存大小至少为size参数所指定的字节数

malloc的返回值是一个指针,指向一段可用内存的起始地址

屡次调用malloc所分配的地址不能有重叠部分,除非某次malloc所分配的地址被释放掉

malloc应该尽快完成内存分配并返回(不能使用NP-hard的内存分配算法)

实现malloc时应同时实现内存大小调整和内存释放函数(realloc和free)




malloc和free函数是配对的,若是申请后不释放就是内存泄露;若是无端释放那就是什么都没有作,释放只能释放一次,若是释放两次及两次以上会出现错误(可是释放空指针例外,释放空指针其实也等于什么都没有作,因此,释放多少次都是能够的)






2,malloc和new

     

     new返回指定类型的指针,而且能够自动计算所须要的大小。

    

      int *p;

      p = new int;   //返回类型为int *类型,分配的大小为sizeof(int)

      p = new int[100];    //返回类型为int *类型,分配的大小为sizeof(int) * 100

    

      而malloc则必须由咱们计算字节数,而且在返回的时候强转成实际指定类型的指针。

  

      int *p;

      p = (int *)malloc(sizeof(int));


          1,malloc的返回是void *,若是咱们写成了: p = malloc(sizeof(int));间接的说明了(将void *转化给了int *,这不合理)

      2,malloc的实参是sizeof(int),用于指明一个整形数据须要的大小,若是咱们写成:

            p =  (int *)malloc(1),          那么能够看出:只是申请了一个字节的空间,若是向里面存放了一个整数的话,

            将会占用额外的3个字节,可能会改变原有内存空间中的数据

     3,malloc只管分配内存,并不能对其进行初始化,因此获得的一片新内存中,其值将是随机的。通常意义上:我

           们习惯性的将其初始化为NULL。            固然,也能够用memset函数的。



简单的说:


malloc 函数其实就是在内存中:找一片指定大小的空间,而后将这个空间的首地址给一个指针变量,这里的指针变量能够是一个单独的指针,也能够是一个数组的首地址, 这要看malloc函数中参数size的具体内容。咱们这里malloc分配的内存空间在逻辑上是连续的,而在物理上能够不连续。咱们做为程序员,关注的 是逻辑上的连续,其它的,操做系统会帮着咱们处理的。

 


下面咱们聊聊malloc的具体实现机制:

Linux内存管理

 

虚拟内存地址与物理内存地址

  为了简单,现代操做系统在处理内存地址时,广泛采用虚拟内存地址技术。即在汇编程序(或机器语言)层面,当涉及内存地址时, 都是使用虚拟内存地址。采用这种技术时,每一个进程仿佛本身独享一片2N字节的内存,其中N是机器位数。例如在64位CPU和64位操做系统下,每一个进程的 虚拟地址空间为264Byte。

  这种虚拟地址空间的做用主要是简化程序的编写及方便操做系统对进程间内存的隔离管理,真实中的进程不太可能(也用不到)如此大的内存空间,实际能用到的内存取决于物理内存大小。

  因为在机器语言层面都是采用虚拟地址,当实际的机器码程序涉及到内存操做时,须要根据当前进程运行的实际上下文将虚拟地址转换为物理内存地址,才能实现对真实内存数据的操做。这个转换通常由一个叫MMU(Memory Management Unit)的硬件完成。

 


页与地址构成

  在现代操做系统中,不管是虚拟内存仍是物理内存,都不是以字节为单位进行管理的,而是以页(Page)为单位。一个内存页是一段固定大小的连续内存地址的总称,具体到Linux中,典型的内存页大小为4096Byte(4K)。

  因此内存地址能够分为页号和页内偏移量。下面以64位机器,4G物理内存,4K页大小为例,虚拟内存地址和物理内存地址的组成以下:

内存地址构成

  上面是虚拟内存地址,下面是物理内存地址。因为页大小都是4K,因此页内便宜都是用低12位表示,而剩下的高地址表示页号。

  MMU映射单位并非字节,而是页,这个映射经过查一个常驻内存的数据结构页表来实现。如今计算机具体的内存地址映射比较复杂,为了加快速度会引入一系列缓存和优化,例如TLB等机制。下面给出一个通过简化的内存地址翻译示意图,虽然通过了简化,可是基本原理与现代计算机真实的状况的一致的。

内存地址翻译


内存页与磁盘页

  咱们知道通常将内存看作磁盘的的缓存,有时MMU在工做时,会发现页表代表某个内存页不在物理内存中,此时会触发一个缺页异 常(Page Fault),此时系统会到磁盘中相应的地方将磁盘页载入到内存中,而后从新执行因为缺页而失败的机器指令。关于这部分,由于能够看作对malloc实现 是透明的,因此再也不详细讲述,有兴趣的能够参考《深刻理解计算机系统》相关章节。

  最后附上一张在维基百科找到的更加符合真实地址翻译的流程供你们参考,这张图加入了TLB和缺页异常的流程(图片来源页)。

较为完整的地址翻译流程


Linux进程级内存管理

  2.2.1 内存排布

  明白了虚拟内存和物理内存的关系及相关的映射机制,下面看一下具体在一个进程内是如何排布内存的。

  以Linux 64位系统为例。理论上,64bit内存地址可用空间为0x0000000000000000 ~ 0xFFFFFFFFFFFFFFFF,这是个至关庞大的空间,Linux实际上只用了其中一小部分(256T)。

  根据Linux内核相关文档描述,Linux64位操做系统仅使用低47位,高17位作扩展(只能是全0或全1)。因此,实际用到的地址为空间为0x0000000000000000 ~ 0x00007FFFFFFFFFFF和0xFFFF800000000000 ~ 0xFFFFFFFFFFFFFFFF,其中前面为用户空间(User Space),后者为内核空间(Kernel Space)。图示以下:

Linux进程地址排布

  对用户来讲,主要关注的空间是User Space。将User Space放大后,能够看到里面主要分为以下几段:

  • Code:这是整个用户空间的最低地址部分,存放的是指令(也就是程序所编译成的可执行机器码)
  • Data:这里存放的是初始化过的全局变量
  • BSS:这里存放的是未初始化的全局变量
  • Heap:堆,这是咱们本文重点关注的地方,堆自低地址向高地址增加,后面要讲到的brk相关的系统调用就是从这里分配内存
  • Mapping Area:这里是与mmap系统调用相关的区域。大多数实际的malloc实现会考虑经过mmap分配较大块的内存区域,本文不讨论这种状况。这个区域自高地址向低地址增加
  • Stack:这是栈区域,自高地址向低地址增加

  下面咱们主要关注Heap区域的操做。对整个Linux内存排布有兴趣的同窗能够参考其它资料。

 

Heap内存模型

  通常来讲,malloc所申请的内存主要从Heap区域分配(本文不考虑经过mmap申请大块内存的状况)。

  由上文知道,进程所面对的虚拟内存地址空间,只有按页映射到物理内存地址,才能真正使用。受物理存储容量限制,整个堆虚拟内存空间不可能所有映射到实际的物理内存。Linux对堆的管理示意以下:

Linux进程堆管理

  Linux维护一个break指针,这个指针指向堆空间的某个地址。从堆起始地址到break之间的地址空间为映射好的,能够供进程访问;而从break往上,是未映射的地址空间,若是访问这段空间则程序会报错。


brk与sbrk

  由上文知道,要增长一个进程实际的可用堆大小,就须要将break指针向高地址移动。Linux经过brk和sbrk系统调用操做break指针。两个系统调用的原型以下:

int brk(void *addr); void *sbrk(intptr_t increment);

  brk将break指针直接设置为某个地址,而sbrk将break从当前位置移动increment所指定的增量。brk 在执行成功时返回0,不然返回-1并设置errno为ENOMEM;sbrk成功时返回break移动以前所指向的地址,不然返回(void *)-1。

  一个小技巧是,若是将increment设置为0,则能够得到当前break的地址。

  另外须要注意的是,因为Linux是按页进行内存映射的,因此若是break被设置为没有按页大小对齐,则系统实际上会在最 后映射一个完整的页,从而实际已映射的内存空间比break指向的地方要大一些。可是使用break以后的地址是很危险的(尽管也许break以后确实有 一小块可用内存地址)。



资源限制与rlimit

  系统对每个进程所分配的资源不是无限的,包括可映射的内存空间,所以每一个进程有一个rlimit表示当前进程可用的资源上限。这个限制能够经过getrlimit系统调用获得,下面代码获取当前进程虚拟内存空间的rlimit:

int main() { struct rlimit *limit = (struct rlimit *)malloc(sizeof(struct rlimit)); getrlimit(RLIMIT_AS, limit); printf("soft limit: %ld, hard limit: %ld\n", limit->rlim_cur, limit->rlim_max); }

  其中rlimit是一个结构体:

struct rlimit { rlim_t rlim_cur; /* Soft limit */ rlim_t rlim_max; /* Hard limit (ceiling for rlim_cur) */ };

  每种资源有软限制和硬限制,而且能够经过setrlimit对rlimit进行有条件设置。其中硬限制做为软限制的上限,非特权进程只能设置软限制,且不能超过硬限制。



实现malloc

  3.1 玩具实现

  在正式开始讨论malloc的实现前,咱们能够利用上述知识实现一个简单但几乎无法用于真实的玩具malloc,权当对上面知识的复习:

/* 一个玩具malloc */ #include <sys/types.h> #include <unistd.h> void *malloc(size_t size) { void *p; p = sbrk(0); if (sbrk(size) == (void *)-1) return NULL; return p; }

  这个malloc每次都在当前break的基础上增长size所指定的字节数,并将以前break的地址返回。这个malloc因为对所分配的内存缺少记录,不便于内存释放,因此没法用于真实场景。



3.2 正式实现

  下面严肃点讨论malloc的实现方案。

  3.2.1 数据结构

  首先咱们要肯定所采用的数据结构。一个简单可行方案是将堆内存空间以块(Block)的形式组织起来,每一个块由meta区和 数据区组成,meta区记录数据块的元信息(数据区大小、空闲标志位、指针等等),数据区是真实分配的内存区域,而且数据区的第一个字节地址即为 malloc返回的地址。

  能够用以下结构体定义一个block:

typedef struct s_block *t_block; struct s_block { size_t size; /* 数据区大小 */ t_block next; /* 指向下个块的指针 */ int free; /* 是不是空闲块 */ int padding; /* 填充4字节,保证meta块长度为8的倍数 */ char data[1] /* 这是一个虚拟字段,表示数据块的第一个字节,长度不该计入meta */ };

  因为咱们只考虑64位机器,为了方便,咱们在结构体最后填充一个int,使得结构体自己的长度为8的倍数,以便内存对齐。示意图以下:

Block结构


关于长度为1的数组注意:http://bbs.csdn.net/topics/300077699


3.2.2 寻找合适的block

  如今考虑如何在block链中查找合适的block。通常来讲有两种查找算法:

  • First fit:从头开始,使用第一个数据区大小大于要求size的块所谓这次分配的块
  • Best fit:从头开始,遍历全部块,使用数据区大小大于size且差值最小的块做为这次分配的块

  两种方法各有千秋,best fit具备较高的内存使用率(payload较高),而first fit具备更好的运行效率。这里咱们采用first fit算法。

/* First fit */ t_block find_block(t_block *last, size_t size) { t_block b = first_block; while(b && !(b->free && b->size >= size)) { *last = b; b = b->next; } return b; }

  find_block从frist_block开始,查找第一个符合要求的block并返回block起始地址,若是找不到 这返回NULL。这里在遍历时会更新一个叫last的指针,这个指针始终指向当前遍历的block。这是为了若是找不到合适的block而开辟新 block使用的,具体会在接下来的一节用到。



3.2.3 开辟新的block

  若是现有block都不能知足size的要求,则须要在链表最后开辟一个新的block。这里关键是如何只使用sbrk建立一个struct:

#define BLOCK_SIZE 24 /* 因为存在虚拟的data字段,sizeof不能正确计算meta长度,这里手工设置 */ t_block extend_heap(t_block last, size_t s) { t_block b; b = sbrk(0); if(sbrk(BLOCK_SIZE + s) == (void *)-1) return NULL; b->size = s; b->next = NULL; if(last) last->next = b; b->free = 0; return b; }

  3.2.4 分裂block

  First fit有一个比较致命的缺点,就是可能会让很小的size占据很大的一块block,此时,为了提升payload,应该在剩余数据区足够大的状况下,将其分裂为一个新的block,示意以下:

分裂block

  实现代码:

void split_block(t_block b, size_t s) { t_block new; new = b->data + s; new->size = b->size - s - BLOCK_SIZE ; new->next = b->next; new->free = 1; b->size = s; b->next = new; }



 

3.2.5 malloc的实现

  有了上面的代码,咱们能够利用它们整合成一个简单但初步可用的malloc。注意首先咱们要定义个block链表的头first_block,初始化为NULL;另外,咱们须要剩余空间至少有BLOCK_SIZE + 8才执行分裂操做。

  因为咱们但愿malloc分配的数据区是按8字节对齐,因此在size不为8的倍数时,咱们须要将size调整为大于size的最小的8的倍数:

size_t align8(size_t s) {
if(s & 0x7 == 0) return s; return ((s >> 3) + 1) << 3; }
#define BLOCK_SIZE 24
void *first_block=NULL; /* other functions... */ void *malloc(size_t size) { t_block b, last; size_t s; /* 对齐地址 */ s = align8(size); if(first_block) { /* 查找合适的block */ last = first_block; b = find_block(&last, s); if(b) { /* 若是能够,则分裂 */ if ((b->size - s) >= ( BLOCK_SIZE + 8)) split_block(b, s); b->free = 0; } else { /* 没有合适的block,开辟一个新的 */ b = extend_heap(last, s); if(!b) return NULL; } } else { b = extend_heap(NULL, s); if(!b) return NULL; first_block = b; } return b->data; }

 


 

3.2.6 calloc的实现

  有了malloc,实现calloc只要两步:

  1. malloc一段内存
  2. 将数据区内容置为0

  因为咱们的数据区是按8字节对齐的,因此为了提升效率,咱们能够每8字节一组置0,而不是一个一个字节设置。咱们能够经过新建一个size_t指针,将内存区域强制看作size_t类型来实现。

void *calloc(size_t number, size_t size) { size_t *new; size_t s8, i; new = malloc(number * size); if(new) { s8 = align8(number * size) >> 3; for(i = 0; i < s8; i++) new[i] = 0; } return new; }

 


 

3.2.7 free的实现

  free的实现并不像看上去那么简单,这里咱们要解决两个关键问题:

  1. 如何验证所传入的地址是有效地址,即确实是经过malloc方式分配的数据区首地址
  2. 如何解决碎片问题

  首先咱们要保证传入free的地址是有效的,这个有效包括两方面:

  • 地址应该在以前malloc所分配的区域内,即在first_block和当前break指针范围内
  • 这个地址确实是以前经过咱们本身的malloc分配的

  第一个问题比较好解决,只要进行地址比较就能够了,关键是第二个问题。这里有两种解决方案:一是在结构体内埋一个magic number字段,free以前经过相对偏移检查特定位置的值是否为咱们设置的magic number,另外一种方法是在结构体内增长一个magic pointer,这个指针指向数据区的第一个字节(也就是在合法时free时传入的地址),咱们在free前检查magic pointer是否指向参数所指地址。这里咱们采用第二种方案:

  首先咱们在结构体中增长magic pointer(同时要修改BLOCK_SIZE):

typedef struct s_block *t_block; struct s_block { size_t size; /* 数据区大小 */ t_block next; /* 指向下个块的指针 */ int free; /* 是不是空闲块 */ int padding; /* 填充4字节,保证meta块长度为8的倍数 */ void *ptr; /* Magic pointer,指向data */ char data[1] /* 这是一个虚拟字段,表示数据块的第一个字节,长度不该计入meta */ };

  而后咱们定义检查地址合法性的函数:

t_block get_block(void *p) { char *tmp; tmp = p; return (p = tmp -= BLOCK_SIZE); } int valid_addr(void *p) { if(first_block) { if(p > first_block && p < sbrk(0)) { return p == (get_block(p))->ptr; } } return 0; }

  当屡次malloc和free后,整个内存池可能会产生不少碎片block,这些block很小,常常没法使用,甚至出现许多碎片连在一块儿,虽然整体能知足某此malloc要求,可是因为分割成了多个小block而没法fit,这就是碎片问题。

  一个简单的解决方式时当free某个block时,若是发现它相邻的block也是free的,则将block和相邻block合并。为了知足这个实现,须要将s_block改成双向链表。修改后的block结构以下:

typedef struct s_block *t_block; struct s_block { size_t size; /* 数据区大小 */ t_block prev; /* 指向上个块的指针 */ t_block next; /* 指向下个块的指针 */ int free; /* 是不是空闲块 */ int padding; /* 填充4字节,保证meta块长度为8的倍数 */ void *ptr; /* Magic pointer,指向data */ char data[1] /* 这是一个虚拟字段,表示数据块的第一个字节,长度不该计入meta */ };

  合并方法以下:

t_block fusion(t_block b) {
if (b->next && b->next->free) { b->size += BLOCK_SIZE + b->next->size; b->next = b->next->next; if(b->next) b->next->prev = b; } return b; }

  有了上述方法,free的实现思路就比较清晰了:首先检查参数地址的合法性,若是不合法则不作任何事;不然,将此block 的free标为1,而且在能够的状况下与后面的block进行合并。若是当前是最后一个block,则回退break指针释放进程内存,若是当前 block是最后一个block,则回退break指针并设置first_block为NULL。实现以下:

void free(void *p) { t_block b; if(valid_addr(p)) { b = get_block(p); b->free = 1; if(b->prev && b->prev->free) b = fusion(b->prev); if(b->next) fusion(b); else { if(b->prev) b->prev->prev = NULL; else first_block = NULL; brk(b); } } }





 3.2.8 realloc的实现

  为了实现realloc,咱们首先要实现一个内存复制方法。如同calloc同样,为了效率,咱们以8字节为单位进行复制:

void copy_block(t_block src, t_block dst) { size_t *sdata, *ddata; size_t i; sdata = src->ptr; ddata = dst->ptr; for(i = 0; (i * 8) < src->size && (i * 8) < dst->size; i++) ddata[i] = sdata[i]; }

  而后咱们开始实现realloc。一个简单(可是低效)的方法是malloc一段内存,而后将数据复制过去。可是咱们能够作的更高效,具体能够考虑如下几个方面:

  • 若是当前block的数据区大于等于realloc所要求的size,则不作任何操做
  • 若是新的size变小了,考虑split
  • 若是当前block的数据区不能知足size,可是其后继block是free的,而且合并后能够知足,则考虑作合并

  下面是realloc的实现:

void *realloc(void *p, size_t size) { size_t s; t_block b, new; void *newp; if (!p) /* 根据标准库文档,当p传入NULL时,至关于调用malloc */ return malloc(size); if(valid_addr(p)) { s = align8(size); b = get_block(p); if(b->size >= s) { if(b->size - s >= (BLOCK_SIZE + 8)) split_block(b,s); } else { /* 看是否可进行合并 */ if(b->next && b->next->free && (b->size + BLOCK_SIZE + b->next->size) >= s) { fusion(b); if(b->size - s >= (BLOCK_SIZE + 8)) split_block(b, s); } else { /* 新malloc */ newp = malloc (s); if (!newp) return NULL; new = get_block(newp); copy_block(b, new); free(p); return(newp); } } return (p); } return NULL; }

  3.3 遗留问题和优化

  以上是一个较为简陋,可是初步可用的malloc实现。还有不少遗留的可能优化点,例如:

  • 同时兼容32位和64位系统
  • 在分配较大快内存时,考虑使用mmap而非sbrk,这一般更高效
  • 能够考虑维护多个链表而非单个,每一个链表中的block大小均为一个范围内,例如8字节链表、16字节链表、24-32字节链表等等。此时能够根据size到对应链表中作分配,能够有效减小碎片,并提升查询block的速度
  • 能够考虑链表中只存放free的block,而不存放已分配的block,能够减小查找block的次数,提升效率

  还有不少可能的优化,这里不一一赘述。下面附上一些参考文献,有兴趣的同窗能够更深刻研究。

  4 其它参考

  1. 这篇文章大量参考了A malloc Tutorial,其中一些图片和代码直接引用了文中的内容,这里特别指出
  2. Computer Systems: A Programmer's Perspective, 2/E一书有许多值得参考的地方
  3. 关于Linux的虚拟内存模型,Anatomy of a Program in Memory是很好的参考资料,另外做者还有一篇How the Kernel Manages Your Memory对于Linux内核中虚拟内存管理的部分有很好的讲解
  4. 对于真实世界的malloc实现,能够参考glibc的实现
  5. 本文写做过程当中大量参考了维基百科,再次感谢这个伟大的网站,而且呼吁你们在手头容许的状况下能够适当捐助维基百科,帮助这个造福人类的系统运行下去
相关文章
相关标签/搜索