OC中swizzling的“移魂大法”

前言

OC的runtime机制是一个很有意思的东西，今天就来讲讲OC方法实现的交换，当然称之为“移魂大法”确实有点夸张了，不过它真的是一个很神奇的东西。

交换的两种方式

1. 在本类中进行方法交换

先贴上代码：

    voidMethodSwizzle(Class c,SEL origSEL,SEL overrideSEL)
    {
        Method origMethod = class_getInstanceMethod(c, origSEL);
        Method overrideMethod= class_getInstanceMethod(c, overrideSEL);
        if(class_addMethod(c, origSEL, method_getImplementation(overrideMethod),method_getTypeEncoding(overrideMethod)))
        {  
             class_replaceMethod(c,overrideSEL, method_getImplementation(origMethod), method_getTypeEncoding(origMethod));
        } else  {
             method_exchangeImplementations(origMethod,overrideMethod);
        }
    }

在本类中交换要考虑两种情况，第一种情况是要复写的方法并没有在目标类中实现，而是在其父类中实现了。第二种情况是这个方法已经存在于目标类中，对于第一种情况，应当先在目标类增加一个新的实现方法(override)，然后将复写的方法替换为原先的实现，运行时函数class_addMethod 如果发现方法已经存在，会失败返回，如果添加成功(在父类中重写的方法)，再把目标类中的方法替换为旧有的实现（译注:addMethod会让目标类的方法指向新的实现，使用replaceMethod再将新的方法指向原先的实现，这样就完成了交换操作。），如果添加失败了，就是第二情况(在目标类重写的方法)。这时可以通过method_exchangeImplementations来完成交换，对于第二种情况，因为class_getInstanceMethod 会返回父类的实现，如果直接替换，就会替换掉父类的实现，而不是目标类中的实现。

2. 两个不同的类交换方法的实现。

    void hookInstanceMethod(Class origClass,SEL origSEL, Class newClass,SEL newSEL)
    {
        Method origMethod = nil;
        Method newMethod = nil;
   
        origMethod = class_getInstanceMethod(origClass, origSEL);
        newMethod = class_getInstanceMethod(newClass, newSEL);
   
        //先在原先的类添加新的Method
        if(class_addMethod(origClass, newSEL, method_getImplementation(origMethod), method_getTypeEncoding(origMethod))){
            method_exchangeImplementations(origMethod, newMethod);
        }
     }
   

对于第二种情况，在旧类中添加一个新方法名，这个新方法名称的实现指向旧方法的实现，这样在旧的类的旧方法的实现(imp)和新类的新方法实现(imp)条换之后，在调用旧方法的时候，这个实际就调用了新的实现，然后要在新的实现中还要调用旧的实现，这时候就是要调用旧类中我们添加的那个新方法名。

实战

上面写了那么多的理论是不是要来点“真枪实弹”的东西，不然扯那么多真的是站着说话腰不疼，那么现在，我拉着你，你跟着我进入一个场景：

在一个像往常一样的早上，你在来上班的路上望着晕沉沉的天空，貌似有种不好的预感要发生什么不好的事情，你到了办公室坐了下来喝了杯咖啡压压惊，这时候产品走了过来跟你讲：我昨天为我们几个流量产品申请了广点通的应用，但是因为有一两个产品流量量比较少，没申请下来，因为广点通的广告会根据Bundle ID来确认对应的应用，所以你那边技术能处理下这个Bundle ID，让不同应用使用同一个广点通的ID吗？好行，这个重任就交给你了。这时你想反驳点什么，但是还没开口话已经被产品说死了，这时你内心的想法：

hit.jpg

这时可能你准备抽出36米长的大刀, 此时产品突然抽出了贴身的长枪。

dou.gif

这时的你没有什么办法了，只能默默的低头按照产品的要求做事情，内心里的世界：

still.gif
开玩笑开玩笑，钱还是要赚的，然后你慢慢整理你扔掉的东西，打开电脑，留下了一个孤独无助的背影...

当然这时候你只要知道runtime 的swizzling，你就有思路了，不就是换个Bundle ID 嘛，是时候展现真正技术的时候了，这时候你打开ida ，将广点通的静态库拉入ida，没办法，你只能慢慢找，看是否有哪个类是专门来获取app信息的，你查看的每个类，F5,F5,F5...，当你点到GDTStatsMgr这个类的时候，这时候你惊奇的发现这个画面：

systemInfo.png
没错，找到类了，这个类就是获取一些app，手机系统信息的类，这时候你查看着每个方法，看哪个方法是获取Bundle ID的，这时候你发现了一个方法initializeData
initializeData.png
这个方法就是用来获取一些基础信息的，你看了下这个类的一些方法，好像没有关于Bundle ID的字段，这时候你发现前面那些方法图里面有个成员变量很奇怪：

-[GDTStatsMgr an]
-[GDTStats setAn:]

嗯，这时你在你的工程delegate里面添加上这句代码：

  hookInstanceMethod(NSClassFromString(@"GDTStatsMgr"),@selector(initializeData),[AppDelegate class],@selector(my_initializeData));

并写上要交换的方法：

 - (void)my_initializeData{
    [self my_initializeData];

    [self setValue:@"想要换成的Bundle ID" forKey:@"an"];
}

OK，这时你颤抖的左手按了下command + R， 这时候一个启动屏广告皓然展现在你的面前, 这时候兴奋的想要大叫一声,不过还是忍了下来,然后淡淡的走过去跟产品说: 你刚才的那个小需求我搞定了。然后转身走的时候留下程序员应有逼格的背影....

总结

OC 中Runtime的swizzling是一个很神奇的东西, 用的好的话可以解决各种疑难杂症,我们一般会在分类中+load写对应的交换, 想要知道为什么,可以点击这里Objective-C Method Swizzling 的最佳实践