如何判断method是否被swizzling

背景

Objective-C的动态性可以让我们干很多事情。比如method swizzling.
但method swizzling有很多负面影响，特别是引入第三方组件后. 如果希望在运行时检测，method是否被swizzling呢？

最近在翻看clang文档时发现可以使用这样一种方法：

代码

static inline bool isDiff(const char *func, SEL _cmd)
{
    char buff[256] = {'\0'};
    if (strlen(func) > 2) {
        char* s = strstr(func, " ") + 1;
        char* e = strstr(func, "]");
        memcpy(buff, s, sizeof(char) * (e - s) );
        return strcmp(buff, sel_getName(_cmd));
    }
    return false;
}

#define ALERT_IF_METHOD_REPLACED assert(!isDiff(__PRETTY_FUNCTION__, _cmd));

解释

__PRETTY_FUNCTION__是一个编译器的宏，在编译期就由编译器决定了，以-[MyObject dosth]为例，在该方法内__PRETTY_FUNCTION__为"-[MyObject dosth]".而_cmd是方法的隐藏参数，是动态的，同样以-[MyObject dosth]为例，_cmd为"dosth"，如果我们执行了一下代码：

Method ori_method = class_getInstanceMethod([MyObject class], @selector(dosth));
Method replace_method = class_getInstanceMethod([MyObject class], @selector(dosth2));
method_exchangeImplementations(ori_method, replace_method);

那么在原来的-[MyObject dosth]函数体中，_cmd为"dosth2"，而__PRETTY_FUNCTION__保持不变。

利用上述特性，前文提到的宏就可以工作了。如下述代码：

Method ori_method = class_getInstanceMethod([MyObject class], @selector(dosth));
Method replace_method = class_getInstanceMethod([MyObject class], @selector(dosth2));
method_exchangeImplementations(ori_method, replace_method);

- (void) dosth
{
    ALERT_IF_METHOD_REPLACED;
}

- (void) dosth2
{
    [self dosth2];
}

总结

然而一般情况下此方法，并没有什么卵用╮(╯_╰)╭　
因为但凡重要的方法，一般是系统方法，如-[UIViewController viewDidLoad],因为无法在该函数体使用上述宏，所以无法检测是否有swizzle。此外如果在dosth2中并没有调用dosth2,那么也没有效果。

但话分两头，在一些大型项目中，使用此方法可用于检测关键方法是否被二方库三方库Hook。另外，为了防止App中关键方法被恶意注入，此方法也很是有参考价值的。

一些参考

这里有一篇讲如何获取当前Method的IMP,戳这里.试了一下，年代比较久远。可用性已经不太好了。╮(╯_╰)╭　
不过。在作者的实现使用了__builtin_return_address(0),获取当前栈的返回地址。这是值得习得的小技能。

后话

后面会继续找一些优雅的检测系统方法被method swizzling的情形。IMP相关的东西比较有意思。持续发现中。

原作写于segmentfault 链接