OC 揭秘：Method Swizzling 的黑魔法（方法交换）

2023-12-12 10:09:52

导言

在 Objective-C 的世界里，有一个鲜为人知的技巧，它拥有改变类方法行为的神奇力量，这就是 Method Swizzling（方法交换）。它是一种动态重写机制，允许我们修改现有类的方法实现，而不必修改源代码本身。

Method Swizzling 的原理

每个 Objective-C 类都拥有一个方法列表，被称为 methodList。methodList 中包含了不同的方法，即 Method，每个 Method 又包含了方法选择器 (SEL) 和方法实现 (IMP)。方法交换的基本原理在于：

断开 SEL 与 IMP 的原始对应关系：
通过修改 methodList 中 Method 的 IMP 指针，我们将 SEL 与原本的 IMP 对应关系打破。
创建新的 SEL 与 IMP 对应关系：
我们将 SEL 与一个新的 IMP 建立对应关系，从而修改方法的行为。

实现 Method Swizzling

Objective-C 运行时 API 提供了实现 Method Swizzling 所需的函数：

class_getInstanceMethod(Class cls, SEL selector)：获取指定类的实例方法的 Method。
class_getClassMethod(Class cls, SEL selector)：获取指定类（类本身）的方法的 Method。
method_setImplementation(Method m, IMP imp)：修改 Method 的 IMP 指针。
method_getImplementation(Method m)：获取 Method 的 IMP 指针。

使用 Class Category 实现 Swizzling

Class Category 提供了一种方便的方式来扩展现有类，而无需修改源代码。我们可以创建 Class Category 来实现 Method Swizzling：

@interface MyClass (Swizzling)

- (void)originalMethod;
- (void)swizzledMethod;

@end

@implementation MyClass (Swizzling)

+ (void)load {
    // 获取原始方法和 swizzled 方法的 Method
    Method originalMethod = class_getInstanceMethod(self, @selector(originalMethod));
    Method swizzledMethod = class_getInstanceMethod(self, @selector(swizzledMethod));
    
    // 交换 IMP 指针
    method_exchangeImplementations(originalMethod, swizzledMethod);
}

@end