• ptmalloc源码分析 - Top chunk的扩容函数sysmalloc实现（09）

    目录

    一、sysmalloc函数基本分配逻辑

    二、强制try_mmap分配方式

    三、非主分配区分配的实现

    1. 设置老的Top chunk的参数

    2. 尝试使用grow_heap函数

    3. 尝试使用new_heap函数

    4. 尝试使用try_mmap方式

    四、主分配区分配的实现

    1. 设置Top扩容的size值

    2. brk分配成功的方式

    3. brk分配失败采用MMAP分配

    五、主分配区分配成功后对齐裁剪操作

    1. 内存地址连续相邻直接扩容Top

    2. brk方式分配地址不相邻情况

    3. mmap方式分配地址不相邻情况

    4. 设置调整后的Topchunk

    六、切割新的Topchunk分配内存

    前面三章，我们讲解了malloc的核心分配函数_int_malloc。_int_malloc核心是从我们管理的bins上去寻找空闲的chunk，并分配内存。如果没有空闲的内存，则到Top chunk上去分配。如果Top chunk也无法满足分配的场景，则需要调用sysmalloc进行内存的分配。

    本章节，我们重点讲sysmalloc的具体实现。

    一、sysmalloc函数基本分配逻辑

    sysmalloc定义：当Top chunk内存空间不足的时候，就会调用sysmalloc函数进行内存分配操作。

    参数：入参nb，为需要分配的；av为分配区状态机对象

    1. /*
    2.    sysmalloc handles malloc cases requiring more memory from the system.
    3.    On entry, it is assumed that av->top does not have enough
    4.    space to service request for nb bytes, thus requiring that av->top
    5.    be extended or replaced.
    6.    调用系统分配函数：sysmalloc
    7.    说明：进入sysmalloc则表示 top chunk的空间不足了，需要进行扩容av->top
    8.    nb：请求的内存大小
    9.    mstate：内存分配状态机(分配区)
    10.  */
    11.  
    12. static void *
    13. sysmalloc (INTERNAL_SIZE_T nb, mstate av)
    14. {
    15.   mchunkptr old_top;              /* 老的Topchunk的地址指针  mstate->top incoming value of av->top */
    16.   INTERNAL_SIZE_T old_size;       /* 老的Top chunk的大小 its size */
    17.   char *old_end;                  /* 老的Top chunk的尾部地址 its end address */
    18.  
    19.   long size;                      /* 第一次分配的内存size arg to first MORECORE or mmap call */
    20.   char *brk;                      /* 通过brk分配后返回的对象 return value from MORECORE */
    21.  
    22.   long correction;                /* 用于记录第二次分配的值 arg to 2nd MORECORE call */
    23.   char *snd_brk;                  /* 第二次处理后返回的值（也是第一次的尾部） 2nd return val */
    24.  
    25.   INTERNAL_SIZE_T front_misalign; /* 不可用的新空间的头部字节 unusable bytes at front of new space */
    26.   INTERNAL_SIZE_T end_misalign;   /* 新内存块尾部对齐的字节 partial page left at end of new space */
    27.   char *aligned_brk;              /* 对齐后的brk值 aligned offset into brk */
    28.  
    29.   mchunkptr p;                    /* 返回的结果p the allocated/returned chunk */
    30.   mchunkptr remainder;            /* Top chunk切割后剩余的remainder chunk remainder from allocation */
    31.   unsigned long remainder_size;   /* Top chunk切割后剩余的remainder chunk 的size its size */
    32.  
    33.  
    34.   size_t pagesize = GLRO (dl_pagesize);
    35.   bool tried_mmap = false;

    sysmalloc基本逻辑：

    1. 非主分配区，通过MMAP并生成heap对象进行Top chunk的扩容操作。
    2. 非主分配区使用MMAP分配，32位操作系统每次分配1M，64位系统每次分配64M
    3. 非主分配区如果分配失败，则调用try_mmap，直接通过MMAP返回需要的内存
    4. 主分配区里面，一般情况下调用brk对Top chunk进行扩容
    5. 主分配区如果通过brk分配失败，则才调用MMAP方式分配
    6. 主分配区brk分配，每次除了需要分配的nb内存字节外，额外还会扩容mp_.top_pad=128K的内存空间；如果主分配区使用MMAP分配，每次分配1M+
    7. 无论主分配区还是非主分配区，分配成功后，都要对分配的内存进行一系列对齐、标记、裁剪、释放等操作
    8. 最后，切割当前需要分配的内存，并将Remainder chunk指向Top chunk地址

    二、强制try_mmap分配方式

    try_mmap是尝试直接进行mmap的方式分配一段内存。当非主分配区通过扩容heap或者new heap都失败的情况下，采用try_mmap方式分配。

    但是需要符合MMAP的阀值以及系统支持MMAP，nb分配的内存为大对象（超过128k）。该场景则直接返回MMAP分配的内存chunk，不调整Top chunk。

    如果MMAP方式分配成功，对分配的内存进行对齐操作，需要计算偏移量，并标记该内存是IS_MMAPPED类型的。最后直接返回内存地址。

    1.   /*
    2.      If have mmap, and the request size meets the mmap threshold, and
    3.      the system supports mmap, and there are few enough currently
    4.      allocated mmapped regions, try to directly map this request
    5.      rather than expanding top.
    6.      1. av==NULL，则直接采用MMAP的方式分配内存
    7.      2. nb分配的内存为大对象（超过128k），并且符合MMAP的阀值以及系统支持MMAP，则采用MMAP分配
    8.      mp_.n_mmaps_max=65536
    9.      mp_.mmap_threshold=128*1024
    11.      需要goto try_mmap，才会进入MMAP分配逻辑，该场景则直接返回MMAP分配的内存chunk，不调整Top chunk
    12.    */
    13.  
    14.   if (av == NULL
    15.       || ((unsigned long) (nb) >= (unsigned long) (mp_.mmap_threshold)
    16. 	  && (mp_.n_mmaps < mp_.n_mmaps_max)))
    17.     {
    18.       char *mm;           /* MMAP的返回值 return value from mmap call*/
    19.  
    20.       /* 尝试进行MMAP分配 */
    21.     try_mmap:
    22.       /*
    23.          Round up size to nearest page.  For mmapped chunks, the overhead
    24.          is one SIZE_SZ unit larger than for normal chunks, because there
    25.          is no following chunk whose prev_size field could be used.
    27.          See the front_misalign handling below, for glibc there is no
    28.          need for further alignments unless we have have high alignment.
    29.        */
    30. 	  /* 进行size的内存对齐操作 */
    31.       if (MALLOC_ALIGNMENT == 2 * SIZE_SZ)
    32.         size = ALIGN_UP (nb + SIZE_SZ, pagesize); //调整size
    33.       else
    34.         size = ALIGN_UP (nb + SIZE_SZ + MALLOC_ALIGN_MASK, pagesize);//调整size
    35.       tried_mmap = true;
    36.  
    37.       /* Don't try if size wraps around 0 */
    38.       if ((unsigned long) (size) > (unsigned long) (nb))
    39.         {
    40.     	  /* 直接调用MMAP映射一块内存，大小为size，可读可写*/
    41.           mm = (char *) (MMAP (0, size, PROT_READ | PROT_WRITE, 0)); //调用MMAP的方式，获取一块size大小的内存
    42.  
    43.           /* 如果分配成功 */
    44.           if (mm != MAP_FAILED)
    45.             {
    46.               /*
    47.                  The offset to the start of the mmapped region is stored
    48.                  in the prev_size field of the chunk. This allows us to adjust
    49.                  returned start address to meet alignment requirements here
    50.                  and in memalign(), and still be able to compute proper
    51.                  address argument for later munmap in free() and realloc().
    52.                */
    53.  
    54.         	  /* 分配的mm内存地址需要进行内存对齐，front_misalign表示对齐的地址前有多少个可以对齐的字节*/
    55.               if (MALLOC_ALIGNMENT == 2 * SIZE_SZ)
    56.                 {
    57.                   /* For glibc, chunk2mem increases the address by 2*SIZE_SZ and
    58.                      MALLOC_ALIGN_MASK is 2*SIZE_SZ-1.  Each mmap'ed area is page
    59.                      aligned and therefore definitely MALLOC_ALIGN_MASK-aligned.  */
    60.                   assert (((INTERNAL_SIZE_T) chunk2mem (mm) & MALLOC_ALIGN_MASK) == 0);
    61.                   front_misalign = 0;
    62.                 }
    63.               else
    64.                 front_misalign = (INTERNAL_SIZE_T) chunk2mem (mm) & MALLOC_ALIGN_MASK;
    65.  
    66.               /* 如果有可以对齐的字节 */
    67.               if (front_misalign > 0)
    68.                 {
    69.  
    70.                   correction = MALLOC_ALIGNMENT - front_misalign; //计算偏移量
    71.                   p = (mchunkptr) (mm + correction); //mm+偏移量后，得到最终返回的chunk的地址
    72.                   //说白了对齐后，将便宜部分的字节当成了一个新的chunk结构，free的时候要一起清理
    73.                   set_prev_size (p, correction); //设置前一个p->mchunk_prev_size；
    74.                   set_head (p, (size - correction) | IS_MMAPPED); //设置p->mchunk_size，并且需要减去偏移量 （size + MMAP映射）
    75.                 }
    76.               else
    77.                 {
    78.             	  /* 这里不需要对齐操作*/
    79.                   p = (mchunkptr) mm; //获取得到chunk的指针地址
    80.                   set_prev_size (p, 0); //这里没有偏移，所以设置前一个mchunk_prev_size为0
    81.                   set_head (p, size | IS_MMAPPED); //设置p->mchunk_size （size + MMAP映射）
    82.                 }
    83.  
    84.               /* update statistics 更新各种状态*/
    85.  
    86.               int new = atomic_exchange_and_add (&mp_.n_mmaps, 1) + 1;
    87.               atomic_max (&mp_.max_n_mmaps, new);
    88.  
    89.               unsigned long sum;
    90.               sum = atomic_exchange_and_add (&mp_.mmapped_mem, size) + size;
    91.               atomic_max (&mp_.max_mmapped_mem, sum);
    92.  
    93.               check_chunk (av, p);
    94.  
    95.               return chunk2mem (p); //这里返回内存地址，非chunk地址
    96.             }
    97.         }
    98.     }

    三、非主分配区分配的实现

    如果是非主分配区，则通过heap_info方式获取堆信息结构，并且通过MMAP分配内存。

    1. 设置老的Top chunk的参数

    brk和snd_brk默认值都设置为失败。brk为第一次分配后的指针地址；snd_brk为第二次分配的地址，同时也是第一次分配内存的尾部地址。

    1.   /* There are no usable arenas and mmap also failed.  */
    2.   /* 如果分配区为空，则不能分配，返回 */
    3.   if (av == NULL)
    4.     return 0;
    5.  
    6.   /* Record incoming configuration of top */
    7.   /* 获取top 的 顶部信息*/
    8.   old_top = av->top; //老的Topchunk的地址
    9.   old_size = chunksize (old_top); //老的Topchunk的size
    10.   old_end = (char *) (chunk_at_offset (old_top, old_size)); //老的Topchunk的尾部地址
    11.  
    12.   /* brk=第一次分配返回值 ；snd_brk=第二次分配返回值 */
    13.   brk = snd_brk = (char *) (MORECORE_FAILURE); //默认设置分配失败

    2. 尝试使用grow_heap函数

    如果剩余的空间不足，首先通过grow_heap尝试heap区的扩容。

    grow_heap函数进行扩容，有分配页大小的限制，分配页的大小要小于HEAP_MAX_SIZE。

    1. /* 如果不是主分配区，则通过heap_info方式获取堆信息结构，并且通过MMAP分配内存 */
    2.   if (av != &main_arena)
    3.     {
    4.       heap_info *old_heap, *heap;
    5.       size_t old_heap_size;
    6.  
    7.       /* First try to extend the current heap. */
    8.       /* 通过heap_for_ptr，获取当前的heap的数据结构 */
    9.       old_heap = heap_for_ptr (old_top);
    10.       old_heap_size = old_heap->size;
    11.  
    12.  
    13.       /**
    14.        * 如果剩余的空间不足，首先通过grow_heap尝试heap区的扩容
    15.        * grow_heap函数进行扩容，有分配页大小的限制，分配页的大小要小于HEAP_MAX_SIZE
    16.        */
    17.       if ((long) (MINSIZE + nb - old_size) > 0
    18.           && grow_heap (old_heap, MINSIZE + nb - old_size) == 0)
    19.         {
    20.     	  /* 扩容方式成功 */
    21.           av->system_mem += old_heap->size - old_heap_size; //变更系统内存记录，增加部分为新的heap的size减去老的heap的值
    22.           set_head (old_top, (((char *) old_heap + old_heap->size) - (char *) old_top)
    23.                     | PREV_INUSE); //设置p->mchunk_size （size + 使用中）
    24.         }

    3. 尝试使用new_heap函数

    扩容失败，则new一个新的heap ，并将av->top指向到新的heap

    new_heap 通过MMAP方式分配一块内存， 32位系统每次映射1M，64位系统每次映射64M。

    1.       /* 扩容失败，则new一个新的heap ；并将av->top指向到新的heap */
    2.       /* new_heap 通过MMAP方式分配一块内存， 32位系统每次映射1M，64位系统每次映射64M*/
    3.       else if ((heap = new_heap (nb + (MINSIZE + sizeof (*heap)), mp_.top_pad)))
    4.         {
    5.           /* Use a newly allocated heap.  */
    6.           heap->ar_ptr = av; //新的heap指向av
    7.           heap->prev = old_heap; //指向前一个老的heap
    8.           av->system_mem += heap->size; //调整分配区的系统内存
    9.           /* Set up the new top.  */
    10.           /* #define top(ar_ptr) ((ar_ptr)->top) */
    11.           top (av) = chunk_at_offset (heap, sizeof (*heap)); //减去heap结构的长度，就能定位到chunk的偏移量，并将new_heap出来的块放置到av->top上
    12.           set_head (top (av), (heap->size - sizeof (*heap)) | PREV_INUSE); //设置p->mchunk_size
    13.  
    14.           /* Setup fencepost and free the old top chunk with a multiple of
    15.              MALLOC_ALIGNMENT in size. */
    16.           /* The fencepost takes at least MINSIZE bytes, because it might
    17.              become the top chunk again later.  Note that a footer is set
    18.              up, too, although the chunk is marked in use. */
    19.           old_size = (old_size - MINSIZE) & ~MALLOC_ALIGN_MASK;
    20.           set_head (chunk_at_offset (old_top, old_size + 2 * SIZE_SZ), 0 | PREV_INUSE);//设置p->mchunk_size （size + PREV使用中）
    21.           if (old_size >= MINSIZE)
    22.             {
    23.               set_head (chunk_at_offset (old_top, old_size), (2 * SIZE_SZ) | PREV_INUSE);//设置p->mchunk_size （size + PREV使用中）
    24.               set_foot (chunk_at_offset (old_top, old_size), (2 * SIZE_SZ)); //设置p->mchunk_prev_size
    25.               set_head (old_top, old_size | PREV_INUSE | NON_MAIN_ARENA);//设置p->mchunk_size （size + PREV使用中 + 非主分配区）
    26.               _int_free (av, old_top, 1); //释放old_top
    27.             }
    28.           else
    29.             {
    30.               set_head (old_top, (old_size + 2 * SIZE_SZ) | PREV_INUSE);//设置p->mchunk_size （size + PREV使用中）
    31.               set_foot (old_top, (old_size + 2 * SIZE_SZ));//设置p->mchunk_prev_size
    32.             }
    33.         }

    4. 尝试使用try_mmap方式

    1.       /* 直接跳转到try_mmap，通过MMAP分配一块内存 */
    2.       else if (!tried_mmap)
    3.         /* We can at least try to use to mmap memory.  */
    4.         goto try_mmap;
    5.     }

    四、主分配区分配的实现

     如果是主分配区，则直接扩容Top chunk（brk和mmap方式都有）。

    1. 设置Top扩容的size值

    size 需要等于 nb（此次分配的容量） + top_pad（每次分配扩展值128K） + MINSIZE对齐字节

    如果主分配区都是通过brk方式分配的，是连续性分配的，可以减去老的Top chunk剩余的old_size值。因为nb大于old_size。size也需要按照pagesize进行页对齐。

    1.   else     /* av == main_arena  如果是主分配区，则直接扩容Top chunk（brk和mmap方式都有）*/
    2.  
    3.  
    4.     { /* Request enough space for nb + pad + overhead */
    5. 	  //# define DEFAULT_TOP_PAD 131072 = 128*1024  mp_.top_pad默认值128K
    6. 	  //mp_.top_pad 初始化或扩展堆的时候需要多申请的内存大小
    7.       size = nb + mp_.top_pad + MINSIZE; //size=分配的内存需要加上 对齐的一些字节；
    8.  
    9.       /*
    10.          If contiguous, we can subtract out existing space that we hope to
    11.          combine with new space. We add it back later only if
    12.          we don't actually get contiguous space.
    13.          contiguous:用于判断是否为连续brk分配，主分配区先减去已经存在的top空间,再向操作系统申请
    14.          如果是连续brk分配，可以减去原有的old_size
    15.          如果是非连续brk分配，后续还会降old_size加回去
    16.          由于old_size小于nb，top_pad又是128K，所以减去之后也有足够的空间存储
    17.        */
    18.       if (contiguous (av))
    19.         size -= old_size;
    20.  
    21.       /*
    22.          Round to a multiple of page size.
    23.          If MORECORE is not contiguous, this ensures that we only call it
    24.          with whole-page arguments.  And if MORECORE is contiguous and
    25.          this is not first time through, this preserves page-alignment of
    26.          previous calls. Otherwise, we correct to page-align below.
    27.        */
    28.       //按照页进行对齐 对齐
    29.       size = ALIGN_UP (size, pagesize);

    2. brk分配成功的方式

    调用MORECORE进行brk的分配。

    1.      //size大于0，然后通过系统调用(sbrk)分配size大小的内存
    2.       if (size > 0)
    3.         {
    4.           brk = (char *) (MORECORE (size));
    5.           LIBC_PROBE (memory_sbrk_more, 2, brk, size);
    6.         }
    7.  
    8.       /* 如果分配成功 */
    9.       if (brk != (char *) (MORECORE_FAILURE))
    10.         {
    11.           /* Call the `morecore' hook if necessary.  */
    12.     	  //分配成功，进行原子操作，调用__after_morecore_hook
    13.           void (*hook) (void) = atomic_forced_read (__after_morecore_hook);
    14.           if (__builtin_expect (hook != NULL, 0))
    15.             (*hook)();
    16.         }

    3. brk分配失败采用MMAP分配

    进入mmap之后，相当于之前的brk被打断了，则此次size需要加上old_size值。

    并通过MMAP方式分配一个size大小的内存块。

    通过set_noncontiguous函数，设置当前分配区的brk分配已经不连续了。

    1.  /* Cannot merge with old top, so add its size back in */
    2.     	  /* 如果是连续的brk分配（主分配区才存在） */
    3.           if (contiguous (av))
    4.             size = ALIGN_UP (size + old_size, pagesize); //调整size大小，分配失败之后又把oldsize给加回去了
    5.  
    6.           /* If we are relying on mmap as backup, then use larger units */
    7.           /* brk分配失败后，使用MMAP方式作为备用方案；如果size小于1M，则使用MMAP的默认值1M */
    8.           if ((unsigned long) (size) < (unsigned long) (MMAP_AS_MORECORE_SIZE))
    9.             size = MMAP_AS_MORECORE_SIZE; //MMAP_AS_MORECORE_SIZ =  1024 * 1024 = 1M
    10.  
    11.           /* Don't try if size wraps around 0 */
    12.           if ((unsigned long) (size) > (unsigned long) (nb))
    13.             {
    14.         	  /* 这里使用MMAP分配1M的内存块；mbrk为MMAP分配后返回的对象 */
    15.               char *mbrk = (char *) (MMAP (0, size, PROT_READ | PROT_WRITE, 0));
    16.  
    17.               /* 如果分配成功，则处理一下 */
    18.               if (mbrk != MAP_FAILED)
    19.                 {
    20.                   /* We do not need, and cannot use, another sbrk call to find end */
    21.                   brk = mbrk; //调整brk的值
    22.                   snd_brk = brk + size; //这里设置到尾部值
    23.  
    24.                   /*
    25.                      Record that we no longer have a contiguous sbrk region.
    26.                      After the first time mmap is used as backup, we do not
    27.                      ever rely on contiguous space since this could incorrectly
    28.                      bridge regions.
    29.                    */
    30.                   set_noncontiguous (av);//主分配区设置不连续的标记，因为这里使用了MMAP分配
    31.                 }
    32.             }
    33.         }

    五、主分配区分配成功后对齐裁剪操作

    主分配区分配成功后，返回brk值为分配的内存地址。

    后续如果是brk连续分配地址是相邻的，则直接扩容Top

    如果非相邻的，则需要将老的Top chunk剩余的空间释放到bins上，并且经过一定裁剪和对齐之后，返回新的Top chunk。

    1. 内存地址连续相邻直接扩容Top

    ​​​​​​​

    1.       /* 分配成功的情况处理 */
    2.       if (brk != (char *) (MORECORE_FAILURE))
    3.         {
    4.           if (mp_.sbrk_base == 0)
    5.             mp_.sbrk_base = brk;
    6.           av->system_mem += size; //调整av的系统内存大小
    7.  
    8.           /*
    9.              If MORECORE extends previous space, we can likewise extend top size.
    10.            */
    11.  
    12.           //判断是否是通过brk分配的用brk分配的，并且地址是连续的，则直接更新Top chunk即可（就是Top chunk的扩容）
    13.           if (brk == old_end && snd_brk == (char *) (MORECORE_FAILURE))
    14.             set_head (old_top, (size + old_size) | PREV_INUSE); //设置p->mchunk_size （size + PREV使用中）

    2. brk方式分配地址不相邻情况

    新分配的内存地址大于原来的top chunk的结束地址，说明地址是不连续的，使用MMAP分配的，则无法直接扩容Top chunk。

    有连续标记，说明是brk方式分配的。这里需要将老的Top的空间以及字节对齐的一些冗余空间进行brk二次扩容，放到第一次扩容的尾部。

    最终我们要返回两个值：aligned_brk（对齐后的值）和snd_brk（二次扩容的值）

    这里要重点说一下：内存地址对齐的目标是让内存分配更加合理和高效，例如chunk指针地址1001，通过对齐后是1024为启始地址。如果跟老的Top不相邻，则需要将老的Top剩余的Size + 内存地址对齐产生的前置内存碎片 重新扩容到新的Top尾部。老的Top剩余的chunk则需要free。

    1.        //新分配的内存地址大于原来的top chunk的结束地址，说明地址是不连续的，使用MMAP分配的，则无法直接扩容Top chunk
    2.           else
    3.             {
    4.               front_misalign = 0;
    5.               end_misalign = 0;
    6.               correction = 0;
    7.               aligned_brk = brk;
    8.  
    9.               /* handle contiguous cases 有连续标记，说明是brk方式分配的 */
    10.               if (contiguous (av))
    11.                 {
    12.                   /* Count foreign sbrk as system_mem.  */
    13.                   if (old_size)
    14.                     av->system_mem += brk - old_end; //Topchunk不连续，需要记录外部不连续内存的大小
    15.  
    16.                   /* 分配的mm内存地址需要进行内存对齐，front_misalign表示对齐的地址前有多少个可以对齐的字节*/
    17.                   front_misalign = (INTERNAL_SIZE_T) chunk2mem (brk) & MALLOC_ALIGN_MASK;
    18.                   if (front_misalign > 0) //如果大于0，说明可以计算这个偏移量，调整对齐后的aligned_brk地址
    19.                     {
    20.                       /*
    21.                          Skip over some bytes to arrive at an aligned position.
    22.                          We don't need to specially mark these wasted front bytes.
    23.                          They will never be accessed anyway because
    24.                          prev_inuse of av->top (and any chunk created from its start)
    25.                          is always true after initialization.
    26.                          跳过一些字节达到对齐的位置。我们不需要特别标记这些浪费的前字节。
    27.                        */
    28.  
    29.                       correction = MALLOC_ALIGNMENT - front_misalign; //对齐可调整的字节
    30.                       aligned_brk += correction; //aligned_brk往后调整几个字节地址，调整后的brk地址
    31.                     }
    32.  
    33.                   /*
    34.                      If this isn't adjacent to existing space, then we will not
    35.                      be able to merge with old_top space, so must add to 2nd request.
    36.                    */
    37.                   //correction=新扩展出来的chunk需要继续扩容的大小 包含brk对齐后头部减少的字节 + 老的top的size
    38.                   correction += old_size;
    39.  
    40.                   /* Extend the end address to hit a page boundary */
    41.                   end_misalign = (INTERNAL_SIZE_T) (brk + size + correction); //对齐后的新的top chunk的尾部地址
    42.                   correction += (ALIGN_UP (end_misalign, pagesize)) - end_misalign; //尾部地址需要进行一次对齐，对齐后是否需要增加字节
    43.  
    44.                   //所以：correction=brk前置的对齐字节 + 老的top的size + 新的brk尾部的对齐字节
    45.                   //aligned_brk：为调整后的新的chunk的起始地址
    46.                   //snd_brk：2次扩容操作的返回值
    47.  
    48.                   assert (correction >= 0);
    49.                   snd_brk = (char *) (MORECORE (correction)); //进行2次扩容操作，snd_brk为第二次扩容操作，扩容的值：correction
    50.  
    51.                   /*
    52.                      If can't allocate correction, try to at least find out current
    53.                      brk.  It might be enough to proceed without failing.
    55.                      Note that if second sbrk did NOT fail, we assume that space
    56.                      is contiguous with first sbrk. This is a safe assumption unless
    57.                      program is multithreaded but doesn't use locks and a foreign sbrk
    58.                      occurred between our first and second calls.
    59.                    */
    60.  
    61.                   if (snd_brk == (char *) (MORECORE_FAILURE))  //扩容失败
    62.                     {
    63.                       correction = 0;
    64.                       snd_brk = (char *) (MORECORE (0)); //这里又变成尾部值
    65.                     }
    66.                   else
    67.                     {
    68.                       /* Call the `morecore' hook if necessary.  */
    69.                       void (*hook) (void) = atomic_forced_read (__after_morecore_hook); //扩容成功
    70.                       if (__builtin_expect (hook != NULL, 0))
    71.                         (*hook)();
    72.                     }
    73.                 }

    3. mmap方式分配地址不相邻情况

    这里主要调整字节对齐和偏移量。mmap方式不进行二次扩容。

    最终我们要返回两个值：aligned_brk（对齐后的值）和snd_brk（不扩容则直接尾部）

    1.  /* handle non-contiguous cases 没有连续标记的Case，使用MMAP分配的情况 */
    2.               else
    3.                 {
    4.                   if (MALLOC_ALIGNMENT == 2 * SIZE_SZ)
    5.                     /* MORECORE/mmap must correctly align */
    6.                     assert (((unsigned long) chunk2mem (brk) & MALLOC_ALIGN_MASK) == 0);
    7.                   else
    8.                     {
    9.                       front_misalign = (INTERNAL_SIZE_T) chunk2mem (brk) & MALLOC_ALIGN_MASK;
    10.                       if (front_misalign > 0)
    11.                         {
    12.                           /*
    13.                              Skip over some bytes to arrive at an aligned position.
    14.                              We don't need to specially mark these wasted front bytes.
    15.                              They will never be accessed anyway because
    16.                              prev_inuse of av->top (and any chunk created from its start)
    17.                              is always true after initialization.
    18.                            */
    19.  
    20.                           aligned_brk += MALLOC_ALIGNMENT - front_misalign; //调整对齐，aligned_brk为调整后的起始地址
    21.                         }
    22.                     }
    23.  
    24.                   /* Find out current end of memory */
    25.                   if (snd_brk == (char *) (MORECORE_FAILURE))
    26.                     {
    27.                       snd_brk = (char *) (MORECORE (0)); //尾部
    28.                     }
    29.                 }

    4. 设置调整后的Topchunk

    1. 设置av->top指向调整后的aligned_brk

    2. 将老的Top chunk上剩余的空间，释放到bins上

    1.         /* Adjust top based on results of second sbrk  前面brk也好，MMAP也好都成功分配之后的操作*/
    2.               if (snd_brk != (char *) (MORECORE_FAILURE))
    3.                 {
    4.                   av->top = (mchunkptr) aligned_brk; //将top指向调整过的aligned_brk地址
    5.                   set_head (av->top, (snd_brk - aligned_brk + correction) | PREV_INUSE); //设置p->mchunk_size （size + PREV使用中）
    6.                   av->system_mem += correction; //系统内存加上correction
    7.  
    8.                   /*
    9.                      If not the first time through, we either have a
    10.                      gap due to foreign sbrk or a non-contiguous region.  Insert a
    11.                      double fencepost at old_top to prevent consolidation with space
    12.                      we don't own. These fenceposts are artificial chunks that are
    13.                      marked as inuse and are in any case too small to use.  We need
    14.                      two to make sizes and alignments work out.
    15.                    */
    16.  
    17.                   /* 合并加工后的的old_top需要回收free */
    18.                   if (old_size != 0)
    19.                     {
    20.                       /*
    21.                          Shrink old_top to insert fenceposts, keeping size a
    22.                          multiple of MALLOC_ALIGNMENT. We know there is at least
    23.                          enough space in old_top to do this.
    24.                          缩小2个字节，并重新设置old_top->mchunk_size，old_size是有足够空间缩小两个字节的
    25.                        */
    26.                       old_size = (old_size - 4 * SIZE_SZ) & ~MALLOC_ALIGN_MASK;
    27.                       set_head (old_top, old_size | PREV_INUSE);
    28.  
    29.                       /*
    30.                          Note that the following assignments completely overwrite
    31.                          old_top when old_size was previously MINSIZE.  This is
    32.                          intentional. We need the fencepost, even if old_top otherwise gets
    33.                          lost.
    34.                        */
    35. 		      set_head (chunk_at_offset (old_top, old_size),
    36. 				(2 * SIZE_SZ) | PREV_INUSE); //设置
    37. 		      set_head (chunk_at_offset (old_top, old_size + 2 * SIZE_SZ),
    38. 				(2 * SIZE_SZ) | PREV_INUSE);
    39.  
    40.                       /* If possible, release the rest. */
    41.                       if (old_size >= MINSIZE)
    42.                         {
    43.                           _int_free (av, old_top, 1); //释放old_top的chunk
    44.                         }
    45.                     }
    46.                 }

    六、切割新的Topchunk分配内存

    如果Top chunk的大小大于分配的值（nb + MINISIZE），则进行切割分配操作，切割出来的分配出去，剩余的remainder变成Top chunk。

    MINISIZE是最小的分配内存大小，在Top上分配至少留足一个MINISIZE的chunk，因为chunk的用户区域使用中的时候会复用下一个chunk的mchunk_prev_size字段的数据空间

    1.   /* finally, do the allocation */
    2.   p = av->top;
    3.   size = chunksize (p); //获取Top chunk的size ：p->mchunk_size
    4.  
    5.   /* check that one of the above allocation paths succeeded */
    6.   /* 如果Top chunk的大小大于分配的值，则进行切割分配操作，切割出来的分配出去，剩余的remainder变成Top chunk*/
    7.   if ((unsigned long) (size) >= (unsigned long) (nb + MINSIZE))
    8.     {
    9.       remainder_size = size - nb;
    10.       remainder = chunk_at_offset (p, nb);
    11.       av->top = remainder;
    12.       set_head (p, nb | PREV_INUSE | (av != &main_arena ? NON_MAIN_ARENA : 0));
    13.       set_head (remainder, remainder_size | PREV_INUSE);
    14.       check_malloced_chunk (av, p, nb);
    15.       return chunk2mem (p);
    16.     }

    至此，我们将malloc函数的实现基本讲解完了，下一章开始讲解free函数。

  • 相关阅读:
    苹果电脑的文件怎么复制到移动硬盘，macbook文件怎么拷贝到移动硬盘
    python循环引用解决 cannot import partially initialize
    物联网边缘-物联网准入或接入安全防护产品及解决方案
    BLCD无刷电机三相电流采样说明与总结
    1413. Minimum Value to Get Positive Step by Step Sum
    TypeScript DOM类型的声明
    一文理清Arbitrum、Arbitrum One、Arbitrum Nitro和Arbitrum Nova的关系
    kafka日志文件详解及生产常见问题总结
    统一系统脆弱性管理平台：让“网络安全漏洞”无处遁形
    visual studio插件开发-Menu
  • 原文地址：https://blog.csdn.net/initphp/article/details/132874817