Objective-C 方法交换实践（三） - Aspects 源码解析

一、类与变量

AspectOptions

typedef NS_OPTIONS(NSUInteger, AspectOptions) {

    AspectPositionAfter   = 0,            /// 原方法之后 (default)

    AspectPositionInstead = 1,            /// 替换原方法

    AspectPositionBefore  = 2,            /// 原方法之前

    AspectOptionAutomaticRemoval = 1 << 3 /// 执行一次后就移除

};

AspectsContainer

存储AspectIdentifier ，总共有三个数组，分别存储 上面的 After Instead Before 的各种 Aspects。

AspectIdentifier

+ (instancetype)identifierWithSelector:(SEL)selector object:(id)object options:(AspectOptions)options block:(id)block error:(NSError **)error;

从创建方法就可以看出，他是用来记录 要替换的对象object，替换的原方法selector，替换的类型options，以及执行的代码block。其中block的参数是 id<AspectInfo>类型的。

AspectInfo

一种是协议名，标记当前 NSinvocation的一些环境

@protocol AspectInfo <NSObject>

/// The instance that is currently hooked.
- (id)instance;

/// The original invocation of the hooked method.
- (NSInvocation *)originalInvocation;

/// All method arguments, boxed. This is lazily evaluated.
- (NSArray *)arguments;

@end

一种是类名，基本上就是实现上面的协议的类。

@interface AspectInfo : NSObject <AspectInfo>
- (id)initWithInstance:(__unsafe_unretained id)instance invocation:(NSInvocation *)invocation;
@property (nonatomic, unsafe_unretained, readonly) id instance;
@property (nonatomic, strong, readonly) NSArray *arguments;
@property (nonatomic, strong, readonly) NSInvocation *originalInvocation;
@end

二、具体过程

1.aspect_isSelectorAllowedAndTrack 检测是否能够swizzle

检测不能是以下方法 @"retain", @"release", @"autorelease", @"forwardInvocation:"

对于"dealloc"方法位置只能是 AspectPositionBefore

被交换的方法是否未实现

如果是metaClass，如果子类已经hook，返回；如果父类已经hook，返回。否者标记hook的 SEL，为父类标记已经hook 的 Child Class。

2.aspect_getContainerForObject

得到当前 select 对应的 AspectsContainer, 不存在就创建，通过关联对象的方式存储在类中

3.AspectIdentifier

创建 AspectIdentifier 并且添加到AspectsContainer里面去

4.aspect_prepareClassAndHookSelector 进行swizzle操作

• 如果已经操作过，返回

• 如果是metaClass，对metaClass 进行 A操作

• 如果实例对象进行过 object_setClass 操作，对 class 进行A操作

• 否则创建一个 subclass ，对其进行 object_setClass 操作，然后对这个 subclass 进行 A 操作，之后把 当前对象的 isa 指向 subclass

A操作是：

替换类的 forwardInvocation 方法为 __ASPECTS_ARE_BEING_CALLED__

为类增加 aliasSelector 指向原 selector

把原 selector 指向 _objc_msgForward 或者_objc_msgForward_stret

当然里面还有一些容错判断

5.__ASPECTS_ARE_BEING_CALLED__ 流程

取出当前对象（object）的 AspectsContainer 和 类（class）的 AspectsContainer ，调用 Before hooks 的一些方法，调用 instead 的方法，调用 after 的方法

对没有instead的情况检测是否有原方法，没有就走原 forwardInvocation方法或者走 doesNotRecognizeSelector 方法