# Aspects > Aspects 核心原理涉及3个技术点: > > - objc_msgForward(触发消息转发机制) > - NSInvocation > - block 的本质 ## 函数指针、指针函数的区别 定义不同: - 函数指针:本质是一个指针,该指针指向一个函数 - 指针函数:本质是一个函数,函数的返回值是一个指针类型 写法不同 - 函数指针:`int (*fun)(int x,int y)` - 指针函数:`int* fun(int x,int y)` 用法不同: - 函数指针 ```c++ typedef int (*FuncPtr)(int, int); int add(int a, int b) { return a + b; } FuncPtr addPtr = add; int result = addPtr(3, 2); ``` - 指针函数 ```c++ int (*getAddFunction())(int, int) { return add; } int add(int a, int b) { return a + b; } int (*addPtr)(int, int) = getAddFunction(); int result = addPtr(3, 2); ``` ## block 本质 block 详细探索步骤请查看这篇文章 [block 底层原理](https://github.com/FantasticLBP/knowledge-kit/blob/master/Chapter1%20-%20iOS/1.89.md)。接下去查看建议版本的分析。 第一步:编写一个基础 block 第二步:用 `xcrun --sdk iphoneos clang -arch arm64 -rewrite-objc -fobjc-arc -fobjc-runtime=ios-8.0.0 ViewController.m -o ViewController-arm64.cpp` 转成 c++ 查看原理 分析:可以发现 block 本质就是结构体,和 OC 对象一样,也有 isa 指针。block 传递进去的方法,被包装成 block 的成员变量,是一个叫做 FuncPtr 的函数指针了。 是不是我们可以按照系统定义,来构造一个 struct,承接一个 block,然后发起调用呢? ```objective-c typedef NS_OPTIONS(int, AspectBlockFlags) { AspectBlockFlagsHasCopyDisposeHelpers = (1 << 25), AspectBlockFlagsHasSignature = (1 << 30) }; typedef struct AspectBlock { __unused Class isa; AspectBlockFlags Flags; __unused int Reserved; void (__unused *invoke)(struct AspectBlock *block, ...); struct { size_t reserved; size_t Block_size; void (*copy)(void *dst, const void *src); void (*dispose)(const void *); } *descriptor; } *AspectBlockRef; void(^printBlock)(NSString *) = ^void(NSString *msg) { NSLog(@"%@", msg); }; printBlock(@"Hello world"); struct AspectBlock *fakeBlock = (__bridge struct AspectBlock *)printBlock; ((void (*)(void *, NSString *))fakeBlock->invoke)(fakeBlock, @"Hello world"); ``` 思考:我们目前已经用自定义的 struct 来承接了 block 并成功执行了。能否用 NSInvocation 来触发 block? ## NSInvocation 触发 block 一个方法需要成功调用并执行需要3要素: - 方法名称 `SEL` - 方法签名(参数个人、参数类型、返回值类型等信息) `Method Type Encoding` - 方法地址、方法实现 `IMP` 如何从自定义的 block 结构体中获取这些信息呢? AspectsIdentifier:每做一次方法交换,都会转换为一次 AspectsIdentifier。是核心逻辑。 以一个例子作为源码探索切入口,点击跳转到源码中 可以看到给 NSObject 添加分类,核心 2 个 API。一个对象方法、一个类方法: ```objective-c + (id)aspect_hookSelector:(SEL)selector withOptions:(AspectOptions)options usingBlock:(id)block error:(NSError **)error; - (id)aspect_hookSelector:(SEL)selector withOptions:(AspectOptions)options usingBlock:(id)block error:(NSError **)error; ``` 内部都走到 `aspect_add` 方法中。其中都会生成 `AspectIdentifier` ,看看是如何生成的? 第一步:生成 block 签名。 第二步:因为我们通过 AOP 给原始方法添加了 block,最后的效果是既可以调用原始方法,又可以调用 block 添加的代码。实现的前提是什么? 比较 block 和 hook 类的方法的签名信息需要 Match。具体逻辑查看注释。 ```shell (lldb) po blockSignature number of arguments = 2 frame size = 224 is special struct return? NO return value: -------- -------- -------- -------- type encoding (v) 'v' flags {} modifiers {} frame {offset = 0, offset adjust = 0, size = 0, size adjust = 0} memory {offset = 0, size = 0} argument 0: -------- -------- -------- -------- type encoding (@) '@?' flags {isObject, isBlock} modifiers {} frame {offset = 0, offset adjust = 0, size = 8, size adjust = 0} memory {offset = 0, size = 8} argument 1: -------- -------- -------- -------- type encoding (@) '@""' flags {isObject} modifiers {} frame {offset = 8, offset adjust = 0, size = 8, size adjust = 0} memory {offset = 0, size = 8} conforms to protocol 'AspectInfo' ``` OC 方法签名和 block 方法签名是有区别的 - 比如在 Aspects 框架中,block 方法签名的参数个数是比 oc 方法参数个数少的。oc 方法自带 `id self, SEL _cmd` 2个参数。 - 都使用相同的类型编码系统,但是 block 签名可能包含额外的信息,例如捕获的变量类型。比如上面的 block 方法签名的最后一个参数 `'@""'`,其中的 `AspectInfo` 就代表捕获的变量类型。 比如2个不带参数的 OC 方法签名和 block 方法签名: - oc 方法签名:`v@:` = `v` + `@` + `:`,返回值 `void`、参数1 `@` 代表对象、参数2 `:` 代表 SEL 类型 - block 方法签名:`v@?` = `v` + `@?`,返回值 `void`、参数1 `@?` 代表既是对象,又是 block ## objc_msgForward 骚操作: - 将待 hook 的方法,和 `objc_msgForward` 进行交换。 `objc_msgForward` 不管对象有没有实现,都会触发消息转发流程 - 此时会走 Runtime 的 NSObject `forwardInvocation` 流程。且 Aspects 将 `forwardInvocation` 方法指向了 `__ASPECTS_ARE_BEING_CALLED__` 方法。 经历这么一波处理,hook 最后都守口到了 `__ASPECTS_ARE_BEING_CALLED__` 方法中。 前面研究过了 `AspectIdentifier` 的逻辑。接下去继续看看后续步骤。 可以看到内部执行 `aspect_prepareClassAndHookSelector`,其内部会调用 `aspect_hookClass`,又会调用 `aspect_swizzleClassInPlace`,最后调用 `aspect_swizzleForwardInvocation` 方法。 ```objective-c static NSString *const AspectsForwardInvocationSelectorName = @"__aspects_forwardInvocation:"; static void aspect_swizzleForwardInvocation(Class klass) { NSCParameterAssert(klass); // If there is no method, replace will act like class_addMethod. IMP originalImplementation = class_replaceMethod(klass, @selector(forwardInvocation:), (IMP)__ASPECTS_ARE_BEING_CALLED__, "v@:@"); if (originalImplementation) { class_addMethod(klass, NSSelectorFromString(AspectsForwardInvocationSelectorName), originalImplementation, "v@:@"); } AspectLog(@"Aspects: %@ is now aspect aware.", NSStringFromClass(klass)); } ``` 该方法将被 hook 对象的 `forwardInvocation:` 方法替换为 `__aspects_forwardInvocation:`。 回归头继续看下面的逻辑。 `class_replaceMethod(klass, selector, aspect_getMsgForwardIMP(self, selector), typeEncoding)` 可以实现将被 hook 类的 hook 方法,替换为 `_objc_msgForward` 或者某些版本下需要的 `_objc_msgForward_stret`。 比如 ViewController 的 viewWillAppear hook 流程就是: `[UIViewController viewWillAppear:]` -> `_objc_msgForwar` -> `[UIViewController forwardInvocation:]` -> `__ASPECTS_ARE_BEING_CALLED__` ## 总结 Aspects 是 Runtime 使用的一个经典库,处理好核心逻辑后,也做了一些黑名单、线程安全等的保护。也有一些类似日志回放功能的处理。