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.jianshu.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.jianshu.com/p/2f05060fa377

阅读原文

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