iOS黑魔法--Runtime Method swizzling

Method Swizzling技术它能够让我们在运行时替换已有方法来实现我们的一些需求。我们都知道方法由两个部分组成，Selector（SEL）相当于一个方法的id；IMP是方法的实现。这样分开的一个便利之处是selector和IMP之间的对应关系可以被改变。这也是方法拦截替换的实现基础。而 Method Swizzling 可以交换两个方法的实现。

Objective-C 提供了以下 API 来动态替换类方法或实例方法的实现：

class_replaceMethod 替换类方法的定义

method_exchangeImplementations 交换 2 个方法的实现

method_setImplementation 设置 1 个方法的实现

3 个 API 的使用场景:

class_replaceMethod, 当需要替换的方法可能有不存在的情况时，可以考虑使用该方法。
method_exchangeImplementations，当需要交换 2 个方法的实现时使用。
method_setImplementation 最简单的用法，当仅仅需要为一个方法设置其实现方式时使用。

应用场景

应用一：拦截系统自带的方法调用Method Swizzling

一般是在load方法中进行，确保最先被调用。+load方法会在Appdelegate的方法之前执行，是最先执行的方法。

使用场景
Method Swizzling 可以重写某个方法而不用继承，同时还可以调用原先的实现。通常的做法是在category中添加一个方法(当然也可以是一个全新的class)。可以通过method_exchangeImplementations这个运行时方法来交换实现。

+ (void)load {
    
    //原来的viewWillAppear方法
    Method viewWillAppear = class_getInstanceMethod([self class], @selector(viewWillAppear:));
    //需要替换的方法
    Method k_viewWillAppear = class_getInstanceMethod([self class], @selector(kViewWillAppear:));
    
    // 如果 swizzling 的是类方法, 采用如下的方式:
    // Class class = object_getClass((id)self);
    // ...
    // Method originalMethod = class_getClassMethod(class, originalSelector);
    // Method swizzledMethod = class_getClassMethod(class, swizzledSelector);
    method_exchangeImplementations(viewWillAppear, k_viewWillAppear);
}

- (void)kViewWillAppear:(BOOL)animated {
    //调用系统的viewWillAppear:
    [self kViewWillAppear:animated];
    NSString *className = NSStringFromClass([self class]);
    NSLog(@"%@ will appear",className);
}

method_setImplementation 可以让我们提供一个新的函数来代替我们要替换的方法。而不是将两个方法的实现做交换。

Person *person = [[Person alloc] init];
Method originalName = class_getInstanceMethod([person class], @selector(getName));
method_setImplementation(originalName, (IMP) my_getName);

应用二：运用Runtime知识替换系统方法避免崩溃

为一些容易造成崩溃的类，在其分类中的 +(void) load方法中完成方法的替换

在自定义的方法中处理掉那些造成崩溃的逻辑即可。

比如我们通过下标获取数组内对象的时候，可能会应为下标越界导致系统崩溃，那么我们可以交换系统方法来避免app的直接崩溃，让我们的app更加健壮

@implementation NSArray (Safe)  
+ (void)load
  {
     
    [NSClassFromString(@"__NSArray0") swapMethod:@selector(objectAtIndex:)
                                   currentMethod:@selector(ls_zeroObjectAtIndex:)];
    
  }

- (id)ls_zeroObjectAtIndex:(NSUInteger)index
{
    if (index >= self.count)
    {
        return nil;
    }
    return [self ls_zeroObjectAtIndex:index];
}

最后编辑于：2020.07.03 09:20:54