【iOS】方法交换

方法交换

method-swizzling是什么

method-swizzling的含义是方法交换,他的主要作用是在运行的时候将一个方法的实现替换为另一个方法的实现,这就是我们说的iOS黑魔法

OC中,利用method-swizzling实现AOP,AOP(面向切面编程)是一种编程思想,区别于OOP。

其中,AOP是面向切面进行提取封装,提取各个模块中的公共部分,提高模块的复用率,降低业务之间的耦合性;而OOP更加倾向于对业务模块的封装,划分出更加清晰的逻辑单元。

在之前学习探索消息流程的时候,我们了解到可以通过SEL方法查找器来查找method方法,而后得到对应的IMP。而方法交换其实就是将SEL与IMP原本的对应断开,并将SEL和新的IMP生成对应关系

这里笔者附上一张看到的图片来解释其关系:

相关API

objc 复制代码
//通过sel获取方法Method
class_getInstanceMethod://获取实例方法
class_getClassMethod://获取类方法

method_getImplementation://获取一个方法的实现
method_setImplementation://设置一个方法的实现
method_getTypeEncoding://获取方法实现的编码类型
class_addMethod://添加方法实现
class_replaceMethod://用一个方法的实现,替换另一个方法的实现,即aIMP 指向 bIMP,但是bIMP不一定指向aIMP
method_exchangeImplementations://交换两个方法的实现,即 aIMP -> bIMP, bIMP -> aIMP

方法交换的风险

下面我们来看看方法交换中会遇到的问题

method-swizzling使用过程中的一次性问题

一次性:method-swizzling写在load方法中,但是load会主动的调用多次,这会导致方法的重复交换,令方法SEL指向又恢复成原来的IMP的问题出现

故而,这里我们可以通过单例模式的原则,令方法交换仅仅执行一次,这里我们需要使用GCD来实现单例,下面举一个例子说明该问题:

objective-c 复制代码
+ (void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        [LGRuntimeTool lg_bestMethodSwizzlingWithClass:self oriSEL:@selector(helloword) swizzledSEL:@selector(lg_studentInstanceMethod)];
    });
}

在当前类中进行方法交换

当我们进行方法交换的时候,必须交换的是当前类中的方法,若是方法在父类中,直接交换会导致父类的方法被错误的修改,这样会影响其所有的子类。

  • 方法交换的隔离性:必须确保交换的目标方法是当前类已实现或动态添加的方法,避免污染父类方法列表。
  • 动态注册的重要性 :通过 class_addMethod 隔离父类实现,保证交换仅作用于当前类及其子类。
  • 风险规避 :在子类分类中操作父类方法需谨慎,推荐使用 method_setImplementationclass_replaceMethod 控制影响范围。

这里我们先来看看若是直接和其父类进行方法交换会引起的后果:

上面的是父类中的方法,下面在子类的分类中方法交换,我们来看看会发生什么

根据断点显示,我们可以发现在子类Student中方法交换之后子类调用最后结果是正确的,但是到了最后一个断点的时候,可以发现程序报错了。我们来看看这是为什么?

这里我们先看子类Student调用personInstanceMethod方法,由于其imp交换成了lg_studentInstanceMethod,而在子类的分类中有该方法,所以不会报错。但是当父类Person中的imp也被交换成了lg_studentInstanceMethod,但我们并没有在父类中实现该方法,即相关的imp无法找到,就会导致程序崩溃掉。

那么我们怎么做就可以让程序不崩溃呢,这里我们可以通过class_addMethod尝试添加要交换的方法,下面先给出示例:

一般交换方法: 交换自己有的方法 -- 走下面 因为自己有意味添加方法失败

交换自己没有实现的方法:

  • 首先第一步:会先尝试给自己添加要交换的方法 :personInstanceMethod (SEL) -> swiMethod(IMP)

  • 然后再将父类的IMP给swizzle personInstanceMethod(imp) -> swizzledSEL

当我们两个方法都没有实现的情况下,就会进入无限递归,导致最后栈溢出:

原因是 栈溢出递归死循环了,那么为什么会发生递归呢?----主要是因为 personInstanceMethod没有实现,然后在方法交换时,始终都找不到oriMethod,然后交换了寂寞,即交换失败,当我们调用personInstanceMethod(oriMethod)时,也就是oriMethod会进入LG中lg_studentInstanceMethod方法,然后这个方法中又调用了lg_studentInstanceMethod,此时的lg_studentInstanceMethod并没有指向oriMethod ,然后导致了自己调自己,即递归死循环

类方法的方法交换

其实类方法的方法交换和实例方法的方法交换差不多,这里我给出一个例子,这里和实例方法的区别其实只是类方法存在元类中

objective-c 复制代码
+ (void)lg_bestClassMethodSwizzlingWithClass:(Class)cls oriSEL:(SEL)oriSEL swizzledSEL:(SEL)swizzledSEL{
    
    if (!cls) NSLog(@"传入的交换类不能为空");

    Method oriMethod = class_getClassMethod([cls class], oriSEL);
    Method swiMethod = class_getClassMethod([cls class], swizzledSEL);
    
    if (!oriMethod) { // 避免动作没有意义
        // 在oriMethod为nil时,替换后将swizzledSEL复制一个不做任何事的空实现,代码如下:
        class_addMethod(object_getClass(cls), oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(swiMethod));
        method_setImplementation(swiMethod, imp_implementationWithBlock(^(id self, SEL _cmd){
            NSLog(@"来了一个空的 imp");
        }));
    }
    
    // 一般交换方法: 交换自己有的方法 -- 走下面 因为自己有意味添加方法失败
    // 交换自己没有实现的方法:
    //   首先第一步:会先尝试给自己添加要交换的方法 :personInstanceMethod (SEL) -> swiMethod(IMP)
    //   然后再将父类的IMP给swizzle  personInstanceMethod(imp) -> swizzledSEL
    //oriSEL:personInstanceMethod

    BOOL didAddMethod = class_addMethod(object_getClass(cls), oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(swiMethod));
    
    if (didAddMethod) {
        class_replaceMethod(object_getClass(cls), swizzledSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod));
    }else{
        method_exchangeImplementations(oriMethod, swiMethod);
    }
    
}

方法交换的应用

方法交换最常用的一个应用是防止数组、字典等越界崩溃 。在iOS中,NSNumberNSArrayNSDictionary这些都是类簇,一个NSArray的实现可能由多个类组成,所以我们必须获取到其"真身"进行交换,直接对NSarray进行操作是无效的。

下面列举了NSArray和NSDictionary本类的类名,可以通过Runtime函数取出本类。

类名 真身
NSArray __NSArrayI
NSMutableArray __NSArrayM
NSDictionary __NSDictionaryI
NSMutableDictionary __NSDictionaryM

这里我以NSArray为例来看看方法交换的应用:

objective-c 复制代码
#import "NSArray+crush.h"
#import "objc/objc-runtime.h"
@implementation NSArray (crush)
+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        NSLog(@"load");
        Method fromMethod = class_getInstanceMethod(objc_getClass("NSConstantArray"), @selector(objectAtIndex:));
        Method toMethod = class_getInstanceMethod(objc_getClass("NSConstantArray"), @selector(new_objectAtIndex:));
            
        method_exchangeImplementations(fromMethod, toMethod);
    });
    
}
- (id)new_objectAtIndex:(NSUInteger)index {
    NSLog(@"new_objectAtIndex");
    if (index >= self.count) {
        // 越界处理
        NSLog(@"Index %lu out of bounds, array count is %lu.", (unsigned long)index, (unsigned long)self.count);
        return nil;
    } else {
        // 正常访问,注意这里调用的是替换后的方法,因为实现已经交换
        return [self new_objectAtIndex:index];
    }
    
}
   
@end

这里我们新建一个NSArray的分类,交换一下objectAtIndexnew_objectAtIndex方法,我们来看看结果:

这里的NSArray类型是NSConstantArray,虽然我们已经越界了但是程序并没有退出,我们打印了一个报错,这样就保证了这个函数的安全.

相关推荐
viperrrrrrrrrr711 分钟前
大数据学习(125)-hive数据分析
大数据·学习
Mac技巧大咖1 小时前
恶意软件清理工具,让Mac电脑安全更简单
安全·macos·恶意软件清理工具
moxiaoran57532 小时前
uni-app学习笔记十八--uni-app static目录简介
笔记·学习·uni-app
Studying 开龙wu3 小时前
机器学习有监督学习sklearn实战二:六种算法对鸢尾花(Iris)数据集进行分类和特征可视化
学习·算法·机器学习
Lester_11014 小时前
嵌入式学习笔记 - STM32 HAL库以及标准库内核以及外设头文件区别问题
笔记·stm32·单片机·学习
Moonnnn.5 小时前
2023年电赛C题——电感电容测量装置
笔记·学习·硬件工程
拾忆-eleven5 小时前
NLP学习路线图(十六):N-gram模型
人工智能·学习·自然语言处理
aa32422432436 小时前
Scratch节日 | 六一儿童节抓糖果
学习·游戏·青少年编程·节日
moxiaoran57536 小时前
uni-app学习笔记十九--pages.json全局样式globalStyle设置
笔记·学习·uni-app
霸王蟹6 小时前
从前端工程化角度解析 Vite 打包策略:为何选择 Rollup 而非 esbuild。
前端·笔记·学习·react.js·vue·rollup·vite