当前位置：首页 > news >正文

【iOS】方法交换

news 来源：原创 2025/7/13 9:43:22

【iOS】方法交换 method-swizzling

文章目录

【iOS】方法交换 method-swizzling
- 前言
- 什么是method-swizzling
- - 相关API
  - 方法交换的风险
  - - 在load方法中保证只加载一次
    - 要在当前类的方法中进行交换
    - 如果方法依赖于cmd
  - 方法交换的应用

前言

之前看过有关于消息转发的内容,这里我们可以简单看一下有关于iOS的黑魔法,方法交换

什么是method-swizzling

method-swizzling这个的意思就是我们可以在运行的时候,将方法的实现替换成另一个方法的一个实现.

在OC中这个原理是为了实现AOP的编程思想,

其中AOP(Aspect Oriented Programming，面向切面编程)是一种编程的思想，区别于OOP（面向对象编程）

面向切面编程（Aspect-Oriented Programming，AOP）是一种通过分离横切关注点（Cross-Cutting Concerns）来增强代码模块化的编程范式。它通过动态注入逻辑的方式，将与核心业务无关的通用功能（如日志、事务、权限控制等）从代码中解耦，从而提升可维护性和复用性.

应用场景:

日志记录:自动记录方法调用参数、执行时间，无需在每个方法中手动写入日志代码
权限控制:在方法调用前检查用户角色，拦截无权限的请求
性能监控:统计方法耗时、数据库查询次数，定位性能瓶颈

每个类都维护着一个方法列表 ，即methodList，methodList中有不同的方法即Method，每个方法中包含了方法的sel和IMP，方法交换就是将sel和imp原本的对应断开，并将sel和新的IMP生成对应关系

方法交换原理

方法交换的风险

在load方法中保证只加载一次

所谓的一次性就是：mehod-swizzling写在load方法中，而load方法会主动调用多次，这样会导致方法的重复交换，使方法sel的指向又恢复成原来的imp的问题

解决方法采用GCD保证只加载一次:

+ (void)load{static dispatch_once_t onceToken;dispatch_once(&onceToken, ^{[LGRuntimeTool lg_bestMethodSwizzlingWithClass:self oriSEL:@selector(helloword) swizzledSEL:@selector(lg_studentInstanceMethod)];});
}

要在当前类的方法中进行交换

被交换的方法必须是当前类的方法，不能是父类的方法，直接把父类的实现拷贝过来不会起作用。父类的方法必须在调用的时候使用，而不是方法交换时使用。方法交换只能作用于当前类的方法，不能影响父类的方法。

我先看一下结果实例:

int main(int argc, const char * argv[]) {@autoreleasepool {//ISA_MASK  0x00007ffffffffff8ULLCJLPerson *person = [CJLPerson alloc];CJLTeacher* teacher = [CJLTeacher alloc];[teacher sayBye];[person sayBye];        }return 0;
}
//方法交换:
+ (void)load {static dispatch_once_t onceToken;dispatch_once(&onceToken, ^{NSLog(@"123456");[self nanxun_MethodSwizzlingWithClass:self.class oriSEL:@selector(sayBye) swizzledSEL:@selector(sayNO)];});
}
- (void)sayNO{//是否会产生递归？--不会产生递归，原因是sayNO 会走 oriIMP，即sayBye的实现中去[self sayNO];NSLog(@"CJLTeacher分类添加的对象方法:%s",__func__);
}

打印结果:

这里的报错也就是发现我们在这个CJLPerson类中没有找到对应的方法,因为我相当于把子类的方法交换到了父类中,父类的方法列表中找不到子类的方法,但是子类可以找到对应的方法,所以问题就是子类不可以和父类交换方法,会导致父类的方法出现问题.

如果要进行交换可以采用下面的方式

通过class_addMethod尝试添加你要交换的方法:

+ (void)nanxun_MethodSwizzlingWithClass:(Class)cls oriSEL:(SEL)oriSEL swizzledSEL:(SEL)swizzledSEL{if (!cls) NSLog(@"传入的交换类不能为空");Method oriMethod = class_getInstanceMethod(cls, oriSEL);Method swiMethod = class_getInstanceMethod(cls, swizzledSEL);// 一般交换方法: 交换自己有的方法 -- 走下面 因为自己有意味添加方法失败// 交换自己没有实现的方法://   首先第一步:会先尝试给自己添加要交换的方法 :personInstanceMethod (SEL) -> swiMethod(IMP)//   然后再将父类的IMP给swizzle  personInstanceMethod(imp) -> swizzledSEL BOOL success = class_addMethod(cls, oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(oriMethod)); // 给原先的SEL添加新方法的methodif (success) {// 自己没有 - 交换 - 没有父类进行处理 (重写一个)class_replaceMethod(cls, swizzledSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod)); } else { // 自己有method_exchangeImplementations(oriMethod, swiMethod);}   
}
//但是这样还不能保证如果两个方法为空不会报错//封装的method-swizzling方法
+ (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:personInstanceMethodBOOL 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);}}

如果两个方法都没有实现会进入无限递归也就是无限循环,导致我们的一个栈溢出:

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

如果方法依赖于cmd

如果方法依赖于cmd,交换之后,cmd就会发生改变会出现很多奇奇怪怪的问题.

方法交换的应用

这里我们可以把它运用解决我们的数组越界的一个问题

在iOS中NSNumber、NSArray、NSDictionary等这些类都是类簇，一个NSArray的实现可能由多个类组成。所以如果想对NSArray进行Swizzling，必须获取到其“真身”进行Swizzling，直接对NSArray进行操作是无效的。

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

//
//  NSArray+crash.m
//  GGDebug
//
//  Created by nanxun on 2025/5/4.
//#import "NSArray+crash.h"
#import "objc/objc-runtime.h"
@implementation NSArray (crash)
+ (void)load {static dispatch_once_t onceToken;dispatch_once(&onceToken, ^{NSLog(@"2222");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(@"123");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类型是NSConstantArray,这里我们已经越界了但是程序还没有退出,但是我们提供了一个报错,保证了这个函数的安全.

这里不可以用字面量的方式访问,不然会出现bug.

如果要修改字面量下标,字面量下标我们要修改这个方法objectAtIndexedSubscript

- (id)new_objectAtIndexedSubscript:(NSUInteger)idx {if (idx >= self.count) {// 越界处理NSLog(@"Index %lu out of bounds, array count is %lu.", (unsigned long)idx, (unsigned long)self.count);return nil;} else {// 正常访问，注意这里调用的是替换后的方法，因为实现已经交换return [self new_objectAtIndexedSubscript:idx];}
}

这样就可以保证他的一个安全: