Method Swizzling中的陷阱

作者：代培
地址：http://daipei.me/posts/trap_in_method_swizzling/
转载请注明出处
我的博客搬家了，新博客地址：daipei.me

这篇文章不是介绍什么是方法交换，这类文章很多，如果你不知道什么是方法交换可以看这篇文章：Method Swizzling

陷阱

方法交换是很危险的东西，会产生一些未知的错误，最近在使用方法交换时就遇到了这样的问题，在我交换了一个系统方法后，在调用这个方法时会在我交换的方法中循环无法跳出。

最终我找到了问题的关键，那就是这个方法可能不存在！

每次iOS版本的更新都会出现一些新的API，这些API在之前的版本中就不存在，而你在交换一个方法时如果不考虑它是否存在，那就会导致严重的错误。

比如我曾经交换过一个方法：

- (void)openURL:(NSURL*)url options:(NSDictionary<NSString *, id> *)options completionHandler:(void (^ __nullable)(BOOL success))completion

这个方法只在iOS10之后有，别人在使用的时候也会先判断这个方法是否会被响应，但是我们看下面的交换代码会先添加这个方法，如果这个方法不存在，那么本来不会响应就变成了会响应，那么在iOS10之前的系统就会进入这个方法，导致死循环。

    Class class = [self class];

    Method method1 = class_getInstanceMethod(class, sel1);
    Method method2 = class_getInstanceMethod(class, sel2);

    BOOL didAddMethod =
    class_addMethod(class,
                    sel1,
                    method_getImplementation(method2),
                    method_getTypeEncoding(method2));

    if (didAddMethod) {
        class_replaceMethod(class,
                            sel2,
                            method_getImplementation(method1),
                            method_getTypeEncoding(method1));
    } else {
        method_exchangeImplementations(method1, method2);
    }

解决方案

解决这个问题也很简单就是交换前做一个判断：

    if (![class instancesRespondToSelector:sel1] || ![class instancesRespondToSelector:sel2]) {
        return ;
    }

如果不响应这个方法，直接返回。

封装

为了不重复写方法交换的代码，也能减少错误，我们可以将其封装，将其放在NSObject的Category中再合适不过了。

NSObject+DPExtension.h

#import <Foundation/Foundation.h>

@interface NSObject (DPExtension)

+ (void)intanceMethodExchangeWithOriginSelector:(SEL)sel1 swizzledSelector:(SEL)sel2;

@end

NSObject+DPExtension.m

#import "NSObject+DPExtension.h"
#import <objc/runtime.h>

@implementation NSObject (DPExtension)

+ (void)intanceMethodExchangeWithOriginSelector:(SEL)sel1 swizzledSelector:(SEL)sel2 {

    Class class = [self class];

    if (![class instancesRespondToSelector:sel1] || ![class instancesRespondToSelector:sel2]) {
        return ;
    }

    Method method1 = class_getInstanceMethod(class, sel1);
    Method method2 = class_getInstanceMethod(class, sel2);

    BOOL didAddMethod =
    class_addMethod(class,
                    sel1,
                    method_getImplementation(method2),
                    method_getTypeEncoding(method2));

    if (didAddMethod) {
        class_replaceMethod(class,
                            sel2,
                            method_getImplementation(method1),
                            method_getTypeEncoding(method1));
    } else {
        method_exchangeImplementations(method1, method2);
    }
}

@end

就是这些^_^