[置顶] 关于自动释放池的讲解与介绍

栏目：综合技术时间：2016-07-14 14:42:56

由于 Objective-C 中的内存管理是1个比较大的话题，所以会分为两篇文章来对内存管理中的1些机制进行剖析，1部份分析自动释放池和 autorelease 方法，另外一部份分析 retain、release 方法的实现和自动援用计数。

写在前面

这篇文章会在源代码层面介绍 Objective-C 中自动释放池，和方法的 autorelease 的具体实现。

从 main 函数开始

main 函数可以说是在全部 iOS 开发中非常不起眼的1个函数，它很好地隐藏在 Supporting Files 文件夹中，却是全部 iOS 利用的入口。

main.m 文件中的内容是这样的： int main(int argc, char * argv[]) { @autoreleasepool { return UIApplicationMain(argc, argv, nil, NSStringFromClass([AppDelegate class])); } }

在这个 @autoreleasepool block 中只包括了1行代码，这行代码将所有的事件、消息全部交给了 UIApplication 来处理，但是这不是本文关注的重点。

需要注意的是：全部 iOS 的利用都是包括在1个自动释放池 block 中的。

@autoreleasepool

@autoreleasepool 究竟是甚么？我们在命令行中使用 clang -rewrite-objc main.m 让编译器重新改写这个文件：

$ clang -rewrite-objc main.m

在生成了1大堆正告以后，当前目录下多了1个 main.cpp 文件

这里删除 main 函数中其他无用的代码。

在这个文件中，有1个非常奇怪的 __AtAutoreleasePool 的结构体，前面的注释写到 /* @autoreleasepool */。也就是说 @autoreleasepool {} 被转换为1个 __AtAutoreleasePool 结构体：

{ __AtAutoreleasePool __autoreleasepool; }

想要弄清楚这行代码的意义，我们要在 main.cpp 中查找名为 __AtAutoreleasePool 的结构体：

struct __AtAutoreleasePool { __AtAutoreleasePool() {atautoreleasepoolobj = objc_autoreleasePoolPush();} ~__AtAutoreleasePool() {objc_autoreleasePoolPop(atautoreleasepoolobj);} void * atautoreleasepoolobj; };

这个结构体会在初始化时调用 objc_autoreleasePoolPush() 方法，会在析构时调用 objc_autoreleasePoolPop 方法。

这表明，我们的 main 函数在实际工作时实际上是这样的：

int main(int argc, const char * argv[]) { { void * atautoreleasepoolobj = objc_autoreleasePoolPush(); // do whatever you want objc_autoreleasePoolPop(atautoreleasepoolobj); } return 0; }

@autoreleasepool 只是帮助我们少写了这两行代码而已，让代码看起来更美观，然后要根据上述两个方法来分析自动释放池的实现。

AutoreleasePool 是甚么

这1节开始分析方法 objc_autoreleasePoolPush 和 objc_autoreleasePoolPop 的实现：

void *objc_autoreleasePoolPush(void) { return AutoreleasePoolPage::push(); } void objc_autoreleasePoolPop(void *ctxt) { AutoreleasePoolPage::pop(ctxt); }

上面的方法看上去是对 AutoreleasePoolPage 对应静态方法 push 和 pop 的封装。

这1小节会依照下面的顺序逐渐解析代码中的内容：

AutoreleasePoolPage 的结构

objc_autoreleasePoolPush 方法

objc_autoreleasePoolPop 方法

AutoreleasePoolPage 的结构

AutoreleasePoolPage 是1个 C++ 中的类：

它在 NSObject.mm 中的定义是这样的：

class AutoreleasePoolPage { magic_t const magic; id *next; pthread_t const thread; AutoreleasePoolPage * const parent; AutoreleasePoolPage *child; uint32_t const depth; uint32_t hiwat; };

magic 用于对当前 AutoreleasePoolPage 完全性的校验
thread 保存了当前页所在的线程

每个自动释放池都是由1系列的 AutoreleasePoolPage 组成的，并且每个 AutoreleasePoolPage 的大小都是 4096 字节（16 进制 0x1000）

#define I386_PGBYTES 4096 #define PAGE_SIZE I386_PGBYTES

双向链表

自动释放池中的 AutoreleasePoolPage 是以双向链表的情势连接起来的：

parent 和 child 就是用来构造双向链表的指针。

自动释放池中的栈

如果我们的1个 AutoreleasePoolPage 被初始化在内存的 0x100816000 ~ 0x100817000 中，它在内存中的结构以下：

其中有 56 bit 用于存储 AutoreleasePoolPage 的成员变量，剩下的 0x100816038 ~ 0x100817000 都是用来存储加入到自动释放池中的对象。

begin() 和 end() 这两个类的实例方法帮助我们快速获得 0x100816038 ~ 0x100817000 这1范围的边界地址。

next 指向了下1个为空的内存地址，如果 next 指向的地址加入1个 object，它就会以下图所示移动到下1个为空的内存地址中：

关于 hiwat 和 depth 在文章中其实不会进行介绍，由于它们其实不影响全部自动释放池的实现，也不在关键方法的调用栈中。

POOL_SENTINEL（哨兵对象）

到了这里，你可能想要知道 POOL_SENTINEL 究竟是甚么，还有它为何在栈中。

首先回答第1个问题： POOL_SENTINEL 只是 nil 的别名。

#define POOL_SENTINEL nil

在每一个自动释放池初始化调用 objc_autoreleasePoolPush 的时候，都会把1个 POOL_SENTINEL push 到自动释放池的栈顶，并且返回这个 POOL_SENTINEL 哨兵对象。

int main(int argc, const char * argv[]) { { void * atautoreleasepoolobj = objc_autoreleasePoolPush(); // do whatever you want objc_autoreleasePoolPop(atautoreleasepoolobj); } return 0; }

上面的 atautoreleasepoolobj 就是1个 POOL_SENTINEL。

而当方法 objc_autoreleasePoolPop 调用时，就会向自动释放池中的对象发送 release 消息，直到第1个 POOL_SENTINEL：

objc_autoreleasePoolPush 方法

了解了 POOL_SENTINEL，我们来重新回顾1下 objc_autoreleasePoolPush 方法：

void *objc_autoreleasePoolPush(void) { return AutoreleasePoolPage::push(); }

它调用 AutoreleasePoolPage 的类方法 push，也非常简单：

static inline void *push() { return autoreleaseFast(POOL_SENTINEL); }

在这里会进入1个比较关键的方法 autoreleaseFast，并传入哨兵对象 POOL_SENTINEL：

static inline id *autoreleaseFast(id obj) { AutoreleasePoolPage *page = hotPage(); if (page && !page->full()) { return page->add(obj); } else if (page) { return autoreleaseFullPage(obj, page); } else { return autoreleaseNoPage(obj); } }

上述方法分3种情况选择不同的代码履行：

有 hotPage 并且当前 page 不满

调用 page->add(obj) 方法将对象添加至 AutoreleasePoolPage 的栈中

有 hotPage 并且当前 page 已满

调用 autoreleaseFullPage 初始化1个新的页

调用 page->add(obj) 方法将对象添加至 AutoreleasePoolPage 的栈中

无 hotPage

调用 autoreleaseNoPage 创建1个 hotPage

调用 page->add(obj) 方法将对象添加至 AutoreleasePoolPage 的栈中

最后的都会调用 page->add(obj) 将对象添加到自动释放池中。

hotPage 可以理解为当前正在使用的 AutoreleasePoolPage。

page->add 添加对象

id *add(id obj) 将对象添加到自动释放池页中：

id *add(id obj) { id *ret = next; *next = obj; next++; return ret; }

笔者对这个方法进行了处理，更方便理解。

这个方法其实就是1个压栈的操作，将对象加入 AutoreleasePoolPage 然后移动栈顶的指针。

autoreleaseFullPage（当前 hotPage 已满）

autoreleaseFullPage 会在当前的 hotPage 已满的时候调用：

static id *autoreleaseFullPage(id obj, AutoreleasePoolPage *page) { do { if (page->child) page = page->child; else page = new AutoreleasePoolPage(page); } while (page->full()); setHotPage(page); return page->add(obj); }

它会从传入的 page 开始遍历全部双向链表，直到：

查找到1个未满的 AutoreleasePoolPage
使用构造器传入 parent 创建1个新的 AutoreleasePoolPage

在查找到1个可使用的 AutoreleasePoolPage 以后，会将该页面标记成 hotPage，然后调动上面分析过的 page->add 方法添加对象。

autoreleaseNoPage（没有 hotPage)

如果当前内存中不存在 hotPage，就会调用 autoreleaseNoPage 方法初始化1个

AutoreleasePoolPage： static id *autoreleaseNoPage(id obj) { AutoreleasePoolPage *page = new AutoreleasePoolPage(nil); setHotPage(page); if (obj != POOL_SENTINEL) { page->add(POOL_SENTINEL); } return page->add(obj); }

既然当前内存中不存在 AutoreleasePoolPage，就要从头开始构建这个自动释放池的双向链表，也就是说，新的 AutoreleasePoolPage 是没有 parent 指针的。

初始化以后，将当前页标记为 hotPage，然后会先向这个 page 中添加1个 POOL_SENTINEL 对象，来确保在 pop 调用的时候，不会出现异常。

最后，将 obj 添加到自动释放池中。

objc_autoreleasePoolPop 方法

一样，回顾1下上面提到的 objc_autoreleasePoolPop 方法：

void objc_autoreleasePoolPop(void *ctxt) { AutoreleasePoolPage::pop(ctxt); }

看起来传入任何1个指针都是可以的，但是在全部工程并没有发现传入其他对象的例子。不过在这个方法中传入其它的指针也是可行的，会将自动释放池释放到相应的位置。

我们1般都会在这个方法中传入1个哨兵对象 POOL_SENTINEL，以下图1样释放对象：

objc-autorelease-pop-stack (1).png

对 objc_autoreleasePoolPop 行动的测试

在继续分析这个方法之前做1个小测试，在 objc_autoreleasePoolPop 传入非哨兵对象，测试1下这个方法的行动。

下面是 main.m 文件中的源代码：

#import int main(int argc, const char * argv[]) { @autoreleasepool { NSString *s = @"Draveness"; [s stringByAppendingString:@"-Suffix"]; } return 0; }

在代码的这1行打1个断点，由于这里会调用 autorelease 方法，将字符串加入自动释放池：

当代码运行到这里时，通过 lldb 打印出当前 hotPage 中的栈内容：

通过 static 方法获得当前 hotPage
打印 AutoreleasePoolPage 中的内容
打印当前 next 指针指向的内容，和之前的内容，⑵时已到了 begin() 位置
使用 print()和 printAll()打印自动释放池中内容

然后将字符串 @"Draveness-Suffix" 的指针传入 pop 方法，测试 pop 方法能否传入非哨兵参数。

再次打印当前 AutoreleasePoolPage 的内容时，字符串已不存在了，这说明向 pop 方法传入非哨兵参数是可行的，只是我们1般不会传入非哨兵对象。

让我们重新回到对 objc_autoreleasePoolPop 方法的分析，也就是 AutoreleasePoolPage::pop 方法的调用：

static inline void pop(void *token) { AutoreleasePoolPage *page = pageForPointer(token); id *stop = (id *)token; page->releaseUntil(stop); if (page->child) { if (page->lessThanHalfFull()) { page->child->kill(); } else if (page->child->child) { page->child->child->kill(); } } }

在这个方法中删除大量无关的代码，和对格式进行了调剂。

该静态方法总共做了3件事情：

使用 pageForPointer 获得当前 token 所在的 AutoreleasePoolPage
调用 releaseUntil 方法释放栈中的对象，直到 stop
调用 child 的 kill 方法

我到现在也不是很清楚为何要根据当前页的不同状态 kill 掉不同 child 的页面。

if (page->lessThanHalfFull()) { page->child->kill(); } else if (page->child->child) { page->child->child->kill(); }

pageForPointer 获得 AutoreleasePoolPage

pageForPointer 方法主要是通过内存地址的操作，获得当前指针所在页的首地址：

static AutoreleasePoolPage *pageForPointer(const void *p) { return pageForPointer((uintptr_t)p); } static AutoreleasePoolPage *pageForPointer(uintptr_t p) { AutoreleasePoolPage *result; uintptr_t offset = p % SIZE; assert(offset >= sizeof(AutoreleasePoolPage)); result = (AutoreleasePoolPage *)(p - offset); result->fastcheck(); return result; }

将指针与页面的大小，也就是 4096 取模，得到当前指针的偏移量，由于所有的 AutoreleasePoolPage 在内存中都是对齐的：

p = 0x100816048

p % SIZE = 0x48

result = 0x100816000

而最后调用的方法 fastCheck() 用来检查当前的 result 是否是1个 AutoreleasePoolPage。

通过检查 magic_t 结构体中的某个成员是不是为 0xA1A1A1A1。

releaseUntil 释放对象

releaseUntil 方法的实现以下：

void releaseUntil(id *stop) { while (this->next != stop) { AutoreleasePoolPage *page = hotPage(); while (page->empty()) { page = page->parent; setHotPage(page); } page->unprotect(); id obj = *--page->next; memset((void*)page->next, SCRIBBLE, sizeof(*page->next)); page->protect(); if (obj != POOL_SENTINEL) { objc_release(obj); } } setHotPage(this); }

它的实现还是很容易的，用1个 while 循环延续释放 AutoreleasePoolPage 中的内容，直到 next 指向了 stop 。

使用 memset 将内存的内容设置成 SCRIBBLE，然后使用 objc_release 释放对象。

kill() 方法

到这里，没有分析的方法就只剩下 kill 了，而它会将当前页面和子页面全部删除：

void kill() { AutoreleasePoolPage *page = this; while (page->child) page = page->child; AutoreleasePoolPage *deathptr; do { deathptr = page; page = page->parent; if (page) { page->unprotect(); page->child = nil; page->protect(); } delete deathptr; } while (deathptr != this); }

autorelease 方法

我们已对自动释放池生命周期有1个比较好的了解，最后需要了解的话题就是 autorelease 方法的实现，先来看1下方法的调用栈：

- [NSObject autorelease] └── id objc_object::rootAutorelease() └── id objc_object::rootAutorelease2() └── static id AutoreleasePoolPage::autorelease(id obj) └── static id AutoreleasePoolPage::autoreleaseFast(id obj) ├── id *add(id obj) ├── static id *autoreleaseFullPage(id obj, AutoreleasePoolPage *page) │ ├── AutoreleasePoolPage(AutoreleasePoolPage *newParent) │ └── id *add(id obj) └── static id *autoreleaseNoPage(id obj) ├── AutoreleasePoolPage(AutoreleasePoolPage *newParent) └── id *add(id obj)

在 autorelease 方法的调用栈中，终究都会调用上面提到的 autoreleaseFast 方法，将当前对象加到 AutoreleasePoolPage 中。

这1小节中这些方法的实现都非常容易，只是进行了1些参数上的检查，终究还要调用 autoreleaseFast 方法：

inline id objc_object::rootAutorelease() { if (isTaggedPointer()) return (id)this; if (prepareOptimizedReturn(ReturnAtPlus1)) return (id)this; return rootAutorelease2(); } __attribute__((noinline,used)) id objc_object::rootAutorelease2() { return AutoreleasePoolPage::autorelease((id)this); } static inline id autorelease(id obj) { id *dest __unused = autoreleaseFast(obj); return obj; }

由于在上面已分析过 autoreleaseFast 方法的实现，这里就不会多说了。

小结

全部自动释放池 AutoreleasePool 的实现和 autorelease 方法都已分析完了，我们再来回顾1下文章中的1些内容：

自动释放池是由 AutoreleasePoolPage 以双向链表的方式实现的

当对象调用 autorelease 方法时，会将对象加入 AutoreleasePoolPage 的栈中

调用 AutoreleasePoolPage::pop 方法会向栈中的对象发送 release 消息

参考资料

深入了解各第3方的源代码分析：iOS-Source-Code-Analyze

What is autoreleasepool? - Objective-C

Using Autorelease Pool Blocks

NSAutoreleasePool

黑幕背后的 Autorelease

------分隔线----------------------------

上一篇 Git学习笔记

下一篇 [VB] VB实现一个窗体的增删改查的demo

------分隔线----------------------------