寻找方法在类中最初的实现

我们知道，如果在分类（category）中重写了类的方法，那么在调用对应方法时，被调用的代码是分类的实现代码。其中的原因并不是“分类的实现覆盖了原类方法的实现”，而是在加载分类方法列表的时候，追加在类方法列表前面。类原有的方法实现还在方法列表中，当在查找时，会先找到分类的实现（如果不太熟悉，还请看深入理解Objective-C：Category）。
知道了分类方法“覆盖”原类方法的实现原理之后，我们是否能够做到，保证我们调用的实现代码是原类的实现，而不是分类的实现？利用 Runtime ，我们可以轻易做到，这里我提供了一种实现方法，仅仅当做实验，可能实际开发没有使用场景。

在此之前需要熟悉一下 Method （类的方法）的结构，到 runtime.h 页面我们能够找到如下代码：

/// An opaque type that represents a method in a class definition.
typedef struct objc_method *Method;

接下来到苹果的 runtime 开源代码中找到 Method 对应的结构体是

struct method_t {
    SEL name;///方法的名字，标识符
    const char *types;///方法签名
    IMP imp;///方法的实现地址
  
    /// 省略
};

我的想法是：新增一个 Method ，当调用该方法时需要改变调用的方式， [obj sel] 变为 [obj pi_performSeleter:sel] 。实现上面有两个问题，

1、如何找到类原有方法的 IMP ？
2、 pi_performSeleter 如何保证通用性？

这里，第二问题 YYKit 作者提供了较好的实现，可以拿来使用（谢谢大神）。接下来看看如何解决第一个问题。

查看 Runtime 提供的 API ，我们能够找到 IMP _Nullable class_getMethodImplementation(Class _Nullable cls, SEL _Nonnull name) ,可以拿来试试，结果会让你失望的，因为这个方法返回的 IMP 是分类的实现（具体源码分析见后续文章），这样的结果也是合情合理的，因为这个 API 在寻找 IMP 的路径就是查找方法的路径，找到的 IMP 就是分类的实现。

如果你知道 Class 的结构，方法存放在 class->getdata()->methods 数组中，寻找方法的实现时，是顺着方法列表找的，最初的实现 IMP 在分类同名方法 IMP 后面，我们可以使用 SEL 倒序寻找，找到的第一个 IMP 就是类原有的实现。接着就是新增一个 Method（新的 SEL ，寻找到的 IMP ，原方法的签名）。
具体的代码如下：

static NSString *const s_pi_selector_prefix = @"pi_origin_";
static SEL pi_aliasForSelector(SEL selector) {
    NSCParameterAssert(selector);
    return NSSelectorFromString([NSString stringWithFormat:@"%@_%@",s_pi_selector_prefix,NSStringFromSelector(selector)]);
}

static void pi_addAliasMethodBaseClassAndSelector(Class cls, SEL sel){
    NSCParameterAssert(cls);
    NSCParameterAssert(sel);
    
    unsigned int count;
    IMP targetIMP = nil; Method targetMethod =nil;
    NSString* needSelName = NSStringFromSelector(sel);
    
    while (cls && (targetMethod == nil)) {
        Method *methods = class_copyMethodList(cls, &count);
        for (int i = count-1; i >= 0; i--) {
            Method method = methods[i];
            NSString *methodName = NSStringFromSelector(method_getName(method));
            if ([methodName isEqualToString:needSelName]) {/// find
                targetIMP = ((struct pi_method_t*)method)->imp;
                targetMethod = method;
                break;
            }
        }
        free(methods);
        if (targetMethod) { /// break
            break;
        }
        cls = class_getSuperclass(cls);
    }
    
    SEL newSEL = pi_aliasForSelector(sel);
    if (targetMethod && targetIMP && !class_respondsToSelector(cls, newSEL)) {
        class_addMethod(cls, newSEL, targetIMP, method_getTypeEncoding(targetMethod));
    }
}

Demo地址：PrimalImplement

为了调用类原有的实现，我们还可以怎么做呢？这里还有两种思路：

1、还可以利用 method_exchangeImplementations 交换分类方法的 IMP 和原有实现的 IMP
2、利用 class_replaceMethod 替换分类的 Method

上面两种实现后可以不用修改调用方法的一般方式 [obj sel] ,但是这样的设计并不好，分类的实现莫名就不起作用了；还有就是如果多个分类重写的一个方法，如何保证调用的一定是类原有的实现。