Method Swizzling 理解

在OC中调用一个方法，其实是向一个对象发送消息，查找消息的唯一依据是selector方法。利用OC的动态特性，可以实现在运行时偷换selector对应的方法实现，达到给方法挂钩的目的，也就是hook。

每个类都有一个方法列表，存放着selector的名字和方法实现的映射关系。IMP又点类似函数指针，指向具体的Method实现。

我们可以使用method_exchangeImplementations来交换2个方法中的IMP。如图：

method_exchangeImplementations交换前

20160120155758696.png

method_exchangeImplementations交换后

20160120155804620.png

示例：假如你用pods引入了一个第三方的框架，但是你发现你需要修改第三方框架类里面的一个私有方法，在这个方法的基础上增加一些自己的逻辑判断。这个时候肯定是不建议直接修改第三方框架的方法了。不然pods升级第三方框架后，岂不是给自己找事情做...

如图所示，pods引入的是MMDrawerController这个第三方框架，需要在原来方法基础上增加一些自己的逻辑判断

66D9D69B-02DD-4694-8310-83AE63B17F2F.png

先新建MMDrawerController的一个类别类如图：

988DE766-004E-4E35-B807-9F8D0BAEA0BC.png

代码如下：

#import "MMDrawerController.h"

@interface MMDrawerController (DYAdd)

@end

#import "MMDrawerController+DYAdd.h"
#import "DY_RootVC.h"
#import "BaseNavigationController.h"
#import 
#import 
@implementation MMDrawerController (DYAdd)
#pragma mark - hook 第三方框架的方法
+ (void)load{
    [super load];
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        //hook方法
        __gbh_tracer_swizzleMethod([self class], @selector(gestureRecognizer:shouldReceiveTouch:), @selector(dyGestureRecognizer:shouldReceiveTouch:));

        __gbh_tracer_swizzleMethod([self class],@selector(isPointContainedWithinCenterViewContentRect:),@selector(dyIsPointContainedWithinCenterViewContentRect:));
    });
}

void __gbh_tracer_swizzleMethod(Class class, SEL originalSelector, SEL swizzledSelector){
    Method originalMethod = class_getInstanceMethod(class, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
    
    BOOL didAddMethod =
    class_addMethod(class,
                    originalSelector,
                    method_getImplementation(swizzledMethod),
                    method_getTypeEncoding(swizzledMethod));
    
    if (didAddMethod) {
        class_replaceMethod(class,
                            swizzledSelector,
                            method_getImplementation(originalMethod),
                            method_getTypeEncoding(originalMethod));
    } else {
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}
//需要hook的替换方法-1
-(BOOL)dyGestureRecognizer:(UIGestureRecognizer *)gestureRecognizer shouldReceiveTouch:(UITouch *)touch{
    if([gestureRecognizer isKindOfClass:[UITapGestureRecognizer class]]){
        if(self.openSide == MMDrawerSideNone){
            return NO;
        }
    }
    //由于方法已经被交换,这里调用的实际上是第三方类的gestureRecognizer:(UIGestureRecognizer *)gestureRecognizer shouldReceiveTouch:(UITouch *)touch方法
    return [self dyGestureRecognizer:gestureRecognizer shouldReceiveTouch:touch];
}
//需要hook的替换方法-2
-(BOOL)dyIsPointContainedWithinCenterViewContentRect:(CGPoint)point{
    //判断是否是首个控制器，这里处理的是只有首个控制器才会相应抽屉的滑动手势
    if (!self.centerViewController) {
        return NO;
    }
    DY_RootVC *root  = (DY_RootVC *)self.centerViewController;
    if (!root || ![root isKindOfClass:[DY_RootVC class]]) {
        return NO;
    }
    BaseNavigationController *nav = (BaseNavigationController *)(root.selectedViewController);
    if (!nav || ![nav isKindOfClass:[BaseNavigationController class]]) {
        return NO;
    }
    if (nav.viewControllers.count > 1) {
        return NO;
    }
    //由于方法已经被交换,这里调用的实际上是原来第三方类的isPointContainedWithinCenterViewContentRect:方法
    return [self dyIsPointContainedWithinCenterViewContentRect:point];
}
#pragma mark - 自己新增方法
- (BOOL)gestureRecognizer:(UIGestureRecognizer *)gestureRecognizer shouldRecognizeSimultaneouslyWithGestureRecognizer:(UIGestureRecognizer *)otherGestureRecognizer
{
    if ([gestureRecognizer isKindOfClass:[UIPanGestureRecognizer class]]) {
        UIPanGestureRecognizer* pan = (UIPanGestureRecognizer*)gestureRecognizer;
        CGPoint p = [pan translationInView:pan.view];
        
        if(fabs(p.y) 0)
        {
            return NO;
        }
        
        return fabs(p.y)

代码注释基本还算详细，这样基本上就达到了既想添加一些自己的逻辑判断，又能继续使用原来的方法，以最少的代码注入来到达自己自定义的需求的目的。