简单分析malloc和free源码实现

最新推荐文章于 2024-08-09 16:09:48 发布

Jimmy1224

最新推荐文章于 2024-08-09 16:09:48 发布

阅读量3k

点赞数 2

分类专栏： C/C++ 文章标签：内存管理 malloc源码 free 不要指定大小 C/C++

本文链接：https://blog.csdn.net/c_base_jin/article/details/96032599

版权

C/C++ 专栏收录该内容

107 篇文章 25 订阅

订阅专栏

在编码过程中，我们往往只关心如何使用系统提供的API，比较容易忽略内在的原理。C/C++编码中，能够安全地管理内存是一项基础考察点。比如

malloc和free一定要配对使用，否则导致内存泄漏
不要free内存两次
指针非空判断
不要跨模块malloc和free内存
野指针问题

而今天就来说说glibc内存malloc和free的源码探究以及说明下为什么free指针时，不需要指定内存块大小。

我是在这个网站找到了malloc和free的源码实现，这里给出其链接，附录给下查找过程。

https://code.woboq.org/userspace/glibc/malloc/malloc.c.html

基础定义

这里主要定义了一个allocation_header结构数组，大小为65536，结构体为记录要分配的实际内存大小和索引。

代码如下：

/* Array of known allocations, to track invalid frees.  */
enum { max_allocations = 65536 };
/*全局数组变量，最大内存分配个数为65536*/
static struct allocation_header *allocations[max_allocations];
/*记录索引*/
static size_t allocation_index;
static size_t deallocation_count;

struct  allocation_header
{
  size_t allocation_index; /*分配索引*/
  size_t allocation_size; /*分配大小*/
};

malloc实现

malloc函数内部调用了malloc_internal(), 从实现上看，malloc支持线程安全。
Alt

malloc_internal 记录了当前内存分配信息，包括实际内存分配大小和索引记录。实际内存分配会比用户申请的要多一个sizeof(allocation_header)大小.

malloc_internal 代码如下：

static void *
malloc_internal (size_t size)
{
  if (allocation_index == max_allocations)
  {
    errno = ENOMEM;
    return NULL;
  }
  
  size_t allocation_size = size + sizeof (struct allocation_header);
  if (allocation_size < size)
  {
    errno = ENOMEM;
    return NULL;
  }
  
  size_t index = allocation_index++;
  //实际内存分配
  void *result = mmap (NULL, allocation_size, PROT_READ | PROT_WRITE,
                       MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
  if (result == MAP_FAILED)
    return NULL;

  allocations[index] = result;
  *allocations[index] = (struct allocation_header)
    {
      .allocation_index = index,
      .allocation_size = allocation_size
    };
    
  //给用户分配的内存块开始地址，跳过头部地址
  return allocations[index] + 1;
}

free实现

我们调用完malloc之后，我们都需要调用free接口才可以，但是我们只需要传入malloc返回的地址即可，大小信息可以从头部信息获取。

free源码如下：

void
free (void *ptr)
{
  if (ptr == NULL)
    return;
  lock ();
  //返回内存分配的头部信息，包含了分配的内存大小信息
  //在get_header中对ptr-1，得到实际的内存分配地址。
  struct allocation_header *header = get_header ("free", ptr);
  //完成内存释放工作
  free_internal ("free", header);
  unlock ();
}