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

我们知道,如果在分类(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] ,但是这样的设计并不好,分类的实现莫名就不起作用了;还有就是如果多个分类重写的一个方法,如何保证调用的一定是类原有的实现。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,922评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,591评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,546评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,467评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,553评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,580评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,588评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,334评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,780评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,092评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,270评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,925评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,573评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,194评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,437评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,154评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352

推荐阅读更多精彩内容