Objective-C中的associated object释放时机问题

假如对象A持有对象B，B作为A的associated object，并且表面上B没有其余被强引用的地方，那么对象A被释放时，对象B肯定会同时释放吗？大部分情况下是，但真有不是的时候。最近实现代码的时候不小心就碰到了这样的特殊情况。

需求

需要监听对象A释放（dealloc）并执行对象A的a方法。此时引入对象B，并作为对象A的associated object。A释放时触发B释放，在B的dealloc方法中执行A的a方法。对象B需要一个指向对象A的属性，并公告为unsafe_unretained（或者assign），由于weak指针此时已经失效了。

示例代码

@interface MyObject1 : NSObject@end@implementation MyObject1- (void)foo {    NSLog(@"success");}@end@interface MyObject2 : NSObject@property (nonatomic, unsafe_unretained) MyObject1 *obj1;@end@implementation MyObject2- (void)dealloc {    [self.obj1 foo];}+ (instancetype)create {    return [[self class] new];}@end@implementation ViewController+ (void)load {    [self fun1];}+ (void)fun1 {    MyObject1 *mo1 = [MyObject1 new];    @synchronized (self) {        MyObject2 *mo2 = [MyObject2 create];        mo2.obj1 = mo1;        objc_setAssociatedObject(mo1, @selector(viewDidLoad), mo2, OBJC_ASSOCIATION_RETAIN_NONATOMIC);    }}@end

问题

运行时出现崩溃，unsafe_unretained指针已经野了，和预期的不一样。堆栈是这样的：

观察崩溃的堆栈，发现mo2对象是被自动释放池释放了。由于mo1对象是在函数退出时就立即释放，这样导致mo1比mo2先被销毁，mo2访问了无效指针导致了崩溃。

这个问题和@synchronized有关系，但目前我还不知道它和arc之间有什么联络。下面给出另一个case，修改一行代码就不会崩溃了：

+ (void)fun2 {    MyObject1 *mo1 = [MyObject1 new];    MyObject2 *mo2 = [MyObject2 create];    @synchronized (self) {        mo2.obj1 = mo1;        objc_setAssociatedObject(mo1, @selector(viewDidLoad), mo2, OBJC_ASSOCIATION_RETAIN_NONATOMIC);    }}

实际上只是把mo2的公告移动到了@synchronized外面，堆栈变成了这样：

这时，mo2的释放发生在调用方法的结束时。

分析

使用Hooper查看汇编代码，观察fun1和fun2的不同。节选出关键部分：
fun1：

fun2：

核心在于：fun1中，创立mo2后调用了retain，fun2中，调用的则是objc_retainAutoreleasedReturnValue。

我们再来看看create方法：

关键的一行在最后，调用了objc_autoreleaseReturnValue。

关于objc_retainAutoreleasedReturnValue和objc_autoreleaseReturnValue，请移步 https://www.songma.com/p/2f05060fa377 。大意是，这两个方法成对出现时，可以优化掉[[obj autorelease] retain]这种骚操作。

结论

在fun1中，因为没有objc_retainAutoreleasedReturnValue，取而代之的是retain，导致对象被放入自动释放池。对于@synchronized为什么会造成不同，我还没有那么深入。

由于全局自动释放池会推迟对象的释放，假如代码非常依赖对象的释放时机则会比较危险。我认为这样做是最保险的，创立一个局部自动释放池，保证局部变量在函数结束时立即释放：

+ (void)fun3 {    MyObject1 *mo1 = [MyObject1 new];    @autoreleasepool {        @synchronized (self) {            MyObject2 *mo2 = [MyObject2 create];            mo2.obj1 = mo1;            objc_setAssociatedObject(mo1, @selector(viewDidLoad), mo2, OBJC_ASSOCIATION_RETAIN_NONATOMIC);        }    }}

参考资料

objc_autoreleaseReturnValue和objc_retainAutoreleasedReturnValue函数对ARC的优化 https://www.songma.com/p/2f05060fa377

本文作者：三豊

阅读原文

本文为云栖社区原创内容，未经允许不得转载。