go 语言是如何运行的-内存分配

https://gocn.vip/topics/10078
在上文中,我们介绍了,从虚拟内存的角度,程序内存大致可以分为 5 个段text、data、bss、stack、heap
其中text段用于程序指令、文字、静态常量
data与bss段用于存储全局变量
stack段用于存储函数调用与函数内的变量,stack段的数据可以被 CPU 快速访问,stack段的大小在运行时是不能增加和减少的,销毁只是通过栈指针的移动来实现的。同时,这也是为什么程序有时候会报错 stack overflow 的原因。
stack段的内存分配是编译器实现的,我们无需关心。同时 stack 通常的大小是有限的。
因此对于大内存的分配,或者想手动创建或释放内存,就只能够对heap段进行操作,这就是俗称的动态分配内存。例如 c 语言中的malloc、calloc、free以及 C++ 中的new、delete
内存的分配属于操作系统级别的操作、因此不管是 cc++ 语言的分配,最后都需要调用操作系统的接口。以 linux 为例,malloc 代码可能调用了操作系统接口mmap分配内存
linux 操作系统提供的内存分配接口如下:
mmap/munmap 映射/释放 指定大小的内存.
brk/sbrk – 改变data段`结束的位置来扩展 heap 段的内存
madvise – 给操作系统建议如何管理内存
set_thread_area/get_thread_area – 操作线程本地存储空间
动态内存分配是操作系统为我们做的事情,其效率直接影响到运行在操作系统上的程序。对于一般的程序来说,例如 c 语言中实现的malloc,最后都是通过调用操作系统的接口来实现的。
动态内存的调度是一个艰难复杂的话题,其要实现的目标包括:
快速分配和释放
内存开销小
使用所有内存
避免碎片化
内存分配的算法包括了:
K&R malloc
Region-based allocator
Buddy allocator
dlmalloc
slab allocator
同时,由于算法解决的目标等不同,还会有不同的变种,其他的目标包括:
内存开销小(例如 buddy 的元数据很大)
良好的内存位置
cpu 核心增加时,扩展性好
并发 malloc / free
GO 语言在进行动态内存分配时,实质调用了上面的操作系统接口。由于 Go 语言并没有调用 c 语言的malloc等函数来分配,组织内存,因此,其必须实现自己的内存组织和调度方式。
GO 语言借鉴了 TCMalloc(Thread-Caching Malloc) 的内存分配方式

TCMalloc(Thread-Caching Malloc)
TCMalloc 是一种内存分配算法,比 GNU C 库中的 malloc 要快 2 倍,正如其名字一样,其是对于每一个线程构建了缓存内存。
TCMalloc 解决了多线程时内存分配的锁竞争问题
TCMalloc 对于小对象的分配非常高效
TCMalloc 的核心思想是将内存划分为多个级别,以减少锁的粒度。在 TCMalloc 内部,内存管理分为两部分:小对象内存(thread memory)和大对象内存(page heap)。
小对象内存管理将内存页分成多个固定大小的可分配的 free 列表。因此,每个线程都会有一个无锁的小对象缓存,这使得在并行程序下分配小对象(<= 32k)非常有效。下图的对象代表的是字节。



分配小对象时



我们将在相同大小的线程本地 free list 中查找,如果有,则从列表中删除第一个对象并返回它
如果 free list 中为空,我们从中央 free list 中获取对象(中央 free list 由所有线程共享),将它们放在线程本地 free list 中,并返回其中一个对象
如果中央 free list 也为空,将从中央页分配器中分配内存页,并将其分割为一组相同大小的对象,并将新对象放在中央 free list 中。和之前一样,将其中一些对象移动到线程本地空闲列表中
大对象内存管理由页集合组成,将其称为页堆(page heap)当分配的对象大于 32K 时,将使用大对象分配方式。



第 k 个 free list 列表是包含 k 大小页的 free list。第 256 个列表比较特殊,是长度大于等于 256 页的 free list。



分配大对象时,对于满足 k 大小页的分配



我们在第 k 个 free list 中查找
如果该 free list 为空,则我们查找下一个更大的 free list,依此类推,最终,如有必要,我们将查找最后一个空闲列表。如果更大的 free list 符合条件,则会进行内存分割以符合当前大小。
如果失败,我们将从操作系统中获取内存。
内存是通过连续页(称为 Spans)的运行来管理的(Go 也根据 Spans 来管理内存)



在 TCMalloc 中,span 有两种状态,已分配或是 free 状态。如果为 free,则 span 是位于页堆列表中的一个。如果已分配,则它要么是已移交给应用程序的大对象,要么是已分成多个小对象的序列。



go 内存分配器最初是基于 TCMalloc 的



go 内存分配
Go allocator 与 TCMalloc 类似,内存的管理由一系列页(spans/mspan 对象)组成,使用(线程/协程)本地缓存并根据内存大小进行划分。
mspan
在 go 语言中,Spans 是 8K 或更大的连续内存区域。可以在runtime/mheap.go中对应的 mspan 结构
type mspan struct {
next *mspan // next span in list, or nil if none
prev *mspan // previous span in list, or nil if none
list *mSpanList // For debugging. TODO: Remove.
startAddr uintptr // address of first byte of span aka s.base()
npages uintptr // number of pages in span
manualFreeList gclinkptr // list of free objects in mSpanManual spans
freeindex uintptr
nelems uintptr // number of object in the span.
allocCache uint64
allocBits *gcBits
gcmarkBits *gcBits
sweepgen uint32
divMul uint16 // for divide by elemsize - divMagic.mul
baseMask uint16 // if non-0, elemsize is a power of 2, & this will get object allocation base
allocCount uint16 // number of allocated objects
spanclass spanClass // size class and noscan (uint8)
state mSpanStateBox // mSpanInUse etc; accessed atomically (get/set methods)
needzero uint8 // needs to be zeroed before allocation
divShift uint8 // for divide by elemsize - divMagic.shift
divShift2 uint8 // for divide by elemsize - divMagic.shift2
elemsize uintptr // computed from sizeclass or from npages
limit uintptr // end of data in span
speciallock mutex // guards specials list
specials *special // linked list of special records sorted by offset.
}



如上图,mspan 是一个双向链接列表对象,其中包含页面的起始地址,它具有的页的数量以及其大小。



mspan 有三种类型,分别是:



idle:没有对象,可以释放回操作系统;或重新用于堆内存;或重新用于栈内存
in use:至少具有一个堆对象,并且可能有更多空间
stack:用于协程栈。可以存在于栈中,也可以存在于堆中,但不能同时存在于两者中。
mcache
Go 像 TCMalloc 一样为每一个 逻辑处理器(P)(Logical Processors) 提供一个本地线程缓存(Local Thread Cache)称作 mcache,所以如果 Goroutine 需要内存可以直接从 mcache 中获取,由于在同一时间只有一个 Goroutine 运行在 逻辑处理器(P)(Logical Processors) 上,所以中间不需要任何锁的参与。mcache 包含所有大小规格的 mspan 作为缓存。



对于每一种大小规格都有两个类型:
scan – 包含指针的对象。
noscan – 不包含指针的对象。
采用这种方法的好处之一就是进行垃圾回收时 noscan 对象无需进一步扫描是否引用其他活跃的对象。
mcentral
mcentral 是被所有逻辑处理器共享的
mcentral 对象收集所有给定规格大小的 span。每一个 mcentral 都包含两个 mspan 的列表:
empty mspanList – 没有空闲对象或 span 已经被 mcache 缓存的 span 列表
nonempty mspanList – 有空闲对象的 span 列表



每一个 mcentral 结构体都维护在 mheap 结构体内。
mheap



Go 使用 mheap 对象管理堆,只有一个全局变量。持有虚拟地址空间。
就上我们从上图看到的:mheap 存储了 mcentral 的数组。这个数组包含了各个的 span 的 mcentral。
central [numSpanClasses]struct {
mcentral mcentral
pad [unsafe.Sizeof(mcentral{})%sys.CacheLineSize]byte
}
由于我们有各个规格的 span 的 mcentral,当一个 mcache 从 mcentral 申请 mspan 时,只需要在独立的 mcentral 级别中使用锁,所以其它任何 mcache 在同一时间申请不同大小规格的 mspan 将互不受影响可以正常申请。
pad 为格外增加的字节。对齐填充(Pad)用于确保 mcentrals 以 CacheLineSize 个字节数分隔,所以每一个 MCentral.lock 都可以获取自己的缓存行(cache line),以避免伪共享(false sharing)问题。



图中对应的free[_MaxMHeapList]mSpanList:一个 spanList 数组。每一个 spanList 中的 mspan 包含 1 ~ 127(_MaxMHeapList - 1)个页。例如,free[3] 是一个包含 3 个页的 mspan 链表。free 表示 free list,表示未分配。对应 busy list。



freelarge mSpanList:一个 mspan 的列表,每一个元素 (mspan) 的页数大于 127,通过 mtreap 结构体管理。busylarge 与之相对应。



在进行内存分配时,go 按照大小分成 3 种对象类



小于 16 个字节的对象 Tiny 类
适用于最大 32 kB 的 Small 类
适用于大对象的 large 类
Small 类会被分为大约有 70 个大小,每一个大小都拥有一个 free list



引入 Tiny 这一微小对象是为了适应小字符串和独立的转义变量。



Tiny 微小对象将几个微小的分配请求组合到一个 16 字节的内存块中



当分配 Tiny 对象时:



查看协程的 mcache 的相应 tiny 槽
根据分配对象的大小,将现有子对象(如果存在)的大小四舍五入为 8、4 或 2 个字节
如果当前分配对象与现有 tiny 子对象适合,请将其放置在此处
如果 tiny 槽未发现合适的块:



查看协程的mcache中相应的mspan
扫描mspan的bitmap以找到可用插槽
如果有空闲插槽,对其进行分配并将其用作新的小型插槽对象(这一切都可以在不获取锁的情况下完成)
如果mspan没有可用插槽:



从mcentral的所需大小类的mspan列表中获得一个新的mspan
如果mspan的列表为空:



从mheap获取内存页以用于 mspan
如果mheap为空或没有足够大的内存页



从操作系统中分配一组新的页(至少 1MB)
Go 会在操作系统分配超大的页(称作 arena),分配大量内存页将分摊与 OS 沟通的成本
small 对象分配与 Tiny 对象类似,



分配和释放大对象直接使用mheap,就像在 TCMalloc 中一样,管理了一组 free list



大对象被四舍五入为页大小(8K)的倍数,在 free list 中查找第 k 个 free list,如果其为空,则继续查找更大的一个 free list,直到第 128 个 free list



如果在第 127 个 free list 中找不到,我们在剩余的大内存页(mspan.freelarge字段)中查找跨度,如果失败,则从操作系统获取



总结
Go 内存管理的一般思想是根据分配对象大小的不同,使用不同的内存结构构建不同的内存缓存级别。
将一个从操作系统接收的连续虚拟内存地址分割为多级缓存来减少锁的使用,同时根据指定的大小分配内存减少内存碎片以提高内存分配的效率和在内存释放之后加快 垃圾回收 的速度
下面是 Go 内存分配的直观表达



参考资料
项目链接
作者知乎
blog
False sharing
Data_segment
linux brk/sbrk
linux mmap/munmap
linux set_thread_area/get_thread_area
TCMalloc : Thread-Caching Malloc
GO MEMORY MANAGEMENT
A visual guide to Go Memory Allocator from scratch (Golang)


Category golang