MethodSwizzling

背景

   有一天,项目里要替换整个项目里一个方法的所有的实现,老大说:“如果是你你怎么办?这就像个面试题”,我在那里回答"继承"、methodSignature、refactor、一堆答案没有一个命中。老大给了我一个methodSwizzing的连接,于是就有了这个文章

问题

一个常见的场景就是,当我们想对大量同样的方法执行某个相同的操作。例如我们想在每个ViewController的viewDidLoad方法中添加某行代码,当然我们可以通过继承、让UIViewController、UINavigationController等都实现这个方法。这样,它们的子类也就会执行这个方法。但是,这样会导致我们需要在各种ViewController中添加代码,而且如果有的子类并没有调用父类的viewDidLoad,那就覆盖不到这个子类。此外,在视图控制器的生命周期,响应事件,绘制视图等场景中,都会由这个需求,这个时候method swizzling 就能够为开发带来很好的作用。

示例代码

 最常见的hook代码

+ (void)swizzMethod:(SEL)origSel altMethod:(SEL)altSel {

        Method origMethod =class_getInstanceMethod(self,origSel);

        Method altMethod = class_getInstanceMethod(self, altSel);

        BOOL didAddMethod = class_addMethod([self class],origSel,         method_getImplementation(altMethod),method_getTypeEncoding(altMethod));

        if (didAddMethod) {

            class_replaceMethod([self class],             altSel,method_getImplementation(origMet),method_getTypeEncoding(origMet))

            } else {

            method_exchangeImplementations(class_getInstanceMethod(self, origSel_),             class_getInstanceMethod(self, altSel_));

          }

}

代码解释

我看到这个代码的时候,除了认得字母外,一无所知😑所以我首先查了下里面的每行代码做了什么。

首先代码中使用到了class_getInstanceMethod(self, origSel)这个方法用来返回一个class中指定的实例方法,如果class或者它的superclass不存在对应的实例方法,就会返回NULL

然后代码中使用了class_addMethod这个方法用来给一个class添加一个指定名称和实现的方法: BOOL class_addMethod(Class cls, SEL name, IMP imp, const char *types);

a)   方法一共有四个参数:class是要添加方法的class,name是方法的name,imp是方法的实现,实现里至少需要两个参数self和_cmd。 types是描述参数类型的字符串构成的一个数组

b)   方法会重写superclass's 实现,但是不会替换这个类里已有的实现

c)   一个Objective-C 方法就是一个至少有两个参数的c方法。例如对于一个给定的方法:

        i. void myMethodIMP(id self, SEL _cmd){ // implementation ....}

d)   我们可以像下面的方法一样动态的将它以resolveThisMethodDynamicall名字添加到class中

        i. class_addMethod([self class], @selector(resolveThisMethodDynamically), (IMP) myMethodIMP, "v@:");

知道了以上两个基础,就可以看到。实际上methodSwizzling是先添加了名称是被替换方法的一个指定的替换后的方法

然后方法调用了一个class_replaceMethod函数,函数共四个参数,第一个参数是要修改的方法所在的类,第二个参数是要替换实现的函数名称,之后分别传入方法实现和方法参数的字符串数组。同时方法会返回被替换函数的之前实现。

a)   如果函数名称指定的方法之前不存在,函数就会像method_add一样为类添加一个方法。

b)   如果函数名称指定的方法存在,函数就像method_setImplementation替换函数的实现

 知道了这点就可以看到,方法替换函数第二步在成功添加了被替换函数同名的替换方法后,会将替换函数名对应的函数实现替换为之前的函数实现。如果添加函数不成功,也就是类之前就存在一个同名的函数实现时会调用method_exchangeImplementations函数,函数接收两个 Method 类型参数。然后交换它们的实现。所以,methodSwizzling实际上相当于将添加后的函数实现和原先的函数实现,也就是IMP指针进行了替换。

上述的代码是写在类本身的,当然我们也可以在category中对某个类进行方法替换,那么需要在load方法中调用上面swizzMethod:(SEL)origSel altMethod:(SEL)altSel相似的函数。只不过其中的实现需要改成先调用两次class_addMethod将被替换函数和替换函数都加入到类中,然后,调用method_exchangeImplementations方法将两个函数的实现进行替换。

注意事项

首先看看load 和 initialize方法:swizzling 应该只在 +load 中完成。

a)   在 Objective-C 的运行时中,每个类有两个方法都会自动调用。+load 是在一个类被初始装载时调用,也就是当代码还没有被load的时候代码从文件夹的代码加载到运行的程序中时候会调用load方法

b)   +initialize 是在应用第一次调用该类的类方法或实例方法前调用的。两个方法都是可选的,并且只有在方法被实现的情况下才会被调用。

只调用一次

swizzling方法会替换所有同名的方法的实现,所以,应该确保这个方法只执行一次,不会发生多次替换的情况,这个时候就可以用dispatch_once,这个可以满足所有的需求,而且 dispatch_once也是初始化一个单例方法的标准方法,但是这种方式进行的更改仅限于method所在的类,比如在UIView的子类B中进行方法替换,那么方法替换将仅限于B及B的子类。

Selector、SEL和IMP的概念

Selector 是一个在运行时被注册(或映射)的C类型字符串。Selector由编译器产生并且在当类被加载进内存时由运行时自动进行名字和实现的映射。Method是用来表示函数定义的类型的一个结构体,IMP是一个函数指针,该方法的第一个参数指向调用方法的自身(即内存中类的实例对象,若是调用类方法,该指针则是指向元类对象 metaclass)。第二个参数是这个方法的名字 selector,该方法的真正参数紧随其后。

防止递归

使用的时候,在交换方法实现后记得要调用原生方法的实现,除非确定可以不用调用原生实现,如果不调用,可能会导致一些底层实现错误例如下面这段代码。

a)   - (void) altSel {

   [self altSel];

}

b)   看起来这里会发生递归调用,但是 实际上因为外层的altSel已经替换成了origSel所以实际上是origSel在调用altSel 但是如果里边换成[self origSel] 就会发生递归调用

防止父类和子类同时替换

应该在替换的方法前增加前缀,如果不在load方法中添加替换,而是在caterory中执行,因为category中的方法是在Runtime加载的时候加到类的MethodList,这就会导致如果altSel重名之后,SEL和IMP不匹配,导致hook的结果不对

被hook的方法应该是类自身的方法,如果把继承的IMP copy到自身上面会存在问题,父类的方法应该在调用的时候使用,不应该swizzling的时候copy到子类,如果父类也hock了方法。子类也hook了方法,父类hook的方法可能由于compileSourc中子类category顺序在父类category之前,而导致只调用子类的hook方法,或者两者都不调用,举个形象的例子, UIView有一个方法,实现叫做A, UIView将方法的实现替换成了B,这个时候,UIView的子类又去替换这个A原先方法名的实现,那么它替换的将是B。也就是UIView的子类就没有A的实现了。

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

推荐阅读更多精彩内容