iOS——面向AOP编程(方法交换)

Posted 化作孤岛的瓜

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了iOS——面向AOP编程(方法交换)相关的知识,希望对你有一定的参考价值。

前言:

本文主要介绍ios中方法交换的各种方式具体使用,帮助新手快速入门并且上手。(毕竟我自己学的时候,找资料挺辛苦的,网上的资料大部分都是互相cv 😭)

本文主要包括以下几个部分:

  • 概念
  • 分类实现方法交换
  • 类之间的方法交换(内部交换)
  • 类之间的方法交换(外部交换)
  • C语言实现的方法交换
  • 方法交换实现全局点击拦截

概念

方法交换:

要理解方法交换的话,要了解一些概念:

Objective-C 是一个动态语言,它有runtime运行时系统来动态得创建类和对象。然后oc的类的结构如图所示:

每一个类都有对应的类方法列表,即methodList,methodList中有不同的方法即Method,其中 :SEL 是指向对外的命名

char * 型 的方法类型,

IMP 是方法指针,指向具体的函数实现。

每个方法中包含了方法的sel和IMP

方法交换的核心方法是exchangeImplementations,通过观察源码可以发现

 方法交换的本质就是将两个方法的sel和imp原本的对应断开,并将sel和新的IMP生成对应关系

分类:

分类的概念比较简单,主要是给ios中的系统类或者我们自定义的类增加扩展方法,类似kotlin的扩展函数和dart的扩展方法。

1.分类实现方法交换

首先,通过分类实现方法交换是比较常见的方式,旨在将系统类的方法替换为我们自己定义的方法,这样可以做一些判空,数组越界,异常处理等操作,这里以NSURL为例子,在我们设置URLWithString增加一个判空处理。

NSURL+Extension.h

#import <Foundation/Foundation.h>
@interface NSURL(ExchangeMethod)
@end

NSURL+Extension.m

#import "NSURL+Extension.h"
#import <objc/runtime.h>
@implementation NSURL (ExchangeMethod)

+(void)load
    //获取系统方法结构体
    Method system = class_getClassMethod([self class], @selector(URLWithString:));
    //获取自己方法结构体
    Method own = class_getClassMethod([self class], @selector(SJUrlWithStr:));
    // 交换方法 系统的 URLWithString 和自己的 SJUrlWithStr
    //交换自己方法和系统方法
    method_exchangeImplementations(system, own);
    //以后再使用 URLWithString 的时候 其实是在调用SJUrlWithStr


+(instancetype)SJUrlWithStr:(NSString *)str
    if (str==nil) 
        NSLog(@"字符串为空");
        return nil;
    
    // 下面的 SJUrlWithStr 其实是使用了 URLWithString
    NSURL *url = [NSURL SJUrlWithStr:str];
    return url;


@end

可以看到首先要获取系统方法结构体,然后获取自定义的方法结构体,最后进行交换。

这里需要注意在自己定义的方法结构体最后要调用一下自己的SJUrlWithStr,看起来是自己调自己,但是因为方法已经被互换了,所以其实是使用了 URLWithString。

最后测试一下功能:

//  1.分类实现方法交换
NSString *str = nil;
NSURL *url = [NSURL URLWithString:str];
NSLog(@"%@", url);

运行:

可以看到被替换的方法成功起作用了,会输出我们加入的语句。 

类之间的方法交换

类与类之间的方法交换,有两种实现方式,第一种是内部实现。

我们定义两个类,TestClass1和TestClass2,他们分别有自己的对象方法- (void)test1,- (void)test2和类方法+ (void)sharedClass1和+ (void)sharedClass2,那么我们可以在TestClass1或者TestClass2中的load方法内部进行方法交换操作。

但是load方法是可能存在执行多次的,所以这里可以通过单例设计原则,使方法交换只执行一次,在OC中可以通过dispatch_once实现单例。

基于此,TestClass2的load方法可以如下设计:

+ (void)load 
  static dispatch_once_t onceToken;
  dispatch_once(&onceToken, ^
    //交换不同类 的  对象方法 (与自定义类TestClass1的test1交换)
    HookUtilsDifferentClassInstanceMethod(NSClassFromString(@"TestClass1"),
                                          [self class], @selector(test1),
                                          @selector(test2));
    //交换不同类 的 对象方法  (与NSArray的containsObject:进行交换)
    // HookUtilsDifferentClassInstanceMethod(NSArray.class, [self class],
    // @selector(containsObject:), @selector(test3:)); 交换不同类 的 类方法
    //友好提示:自定义的方法可以写在任何的自定义类中
    HookUtilsDifferentClassClassMethod(NSClassFromString(@"TestClass1"),
                                       [self class], @selector(sharedClass1),
                                       @selector(sharedClass2));
  );

中间注掉了一段,是和系统类的方法交换尝试,测试也是可行的。

测试验证:

//  2.交换不同的类 load实现
[TestClass1 sharedClass1];
TestClass1 *testClass1 = [[TestClass1 alloc] init];
[testClass1 test1];

结果:

可以看到,我们只调用了TestClass1的类方法和对象方法,但是TestClass2的也同样被调用到了。

看到这里,可以发现,只要拿到两个类的类对象,就可以拿到他们各自的方法并进行方法交换,所以从外部去操作两个类,让他们进行方法交换,是可以抽象成通用的方法的:

/**
@brief 交换不同类中两个  对象方法    友好提示:自定义的方法可以写在任何的自定义类中 )
@param originalCls 被交换的类
@param swizzledCls 用来交换的类
@param originalSelector 被交换的方法
@param swizzledSelector 用来交换的方法
  */
void HookUtilsDifferentClassInstanceMethod(Class originalCls,Class swizzledCls,SEL originalSelector, SEL swizzledSelector) 
    if (!originalCls || !swizzledCls) 
        NSLog(@"交换方法失败--请保证交换的类名不为空");
        return;
    
    if (!originalSelector || !swizzledSelector) 
        NSLog(@"交换方法失败--请保证交换的方法名不为空");
        return;
    
    //通过class_getClassMethod 获取两个方法Method
    Method originalMethod = class_getInstanceMethod(originalCls, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(swizzledCls, swizzledSelector);
    //交换之前,先对自定义方法进行添加
    BOOL didAddMethod = class_addMethod(originalCls,
                                        swizzledSelector,
                                        method_getImplementation(swizzledMethod),
                                        method_getTypeEncoding(swizzledMethod));
    //如果添加成功,则进行交换
    if (didAddMethod) 
        method_exchangeImplementations(originalMethod, class_getInstanceMethod(originalCls, swizzledSelector));
     else 
        NSLog(@"交换方法失败--添加方法失败");
    

这里有个很重要的判断 didAddMethod

为什么要做这个判断呢?

因为如果目标类里面的originalCls没有实现(子类没有实现,父类实现了,所以可以调用但是空实现),然后在方法交换时,始终都找不到oriMethod(原方法),然后交换了寂寞,即交换失败。这样当我们最后去调用被交换的方法的时候,目标类内部就找不到oriMethod,就会变成swizzledMethod(待交换方法)自己无限调自己,导致栈溢出。

C语言实现的方法交换

可以通过C语言实现交换,这样做的好处是可以尽量避免被AppStore监测到(我没有试过。。存疑~-~)。因为JSPatch的原因,苹果封杀了所有带有热更新功能的代码,所以方法交换其实是要慎重使用的,避免被判断为热更新而导致app下架。

C语言实现功能类似,就不赘述了,大家看代码吧:

CHookUtils.h

#import <Foundation/Foundation.h>
#import <objc/runtime.h>
#include <objc/message.h>
#ifdef __cplusplus
extern "C" 
#endif

bool OBJC_EXCHANGE_METHOD_SAFE(const char* orgClassStr,
                               const char* orgSelector,
                               const char* newClassStr,
                               const char* newSelector);

bool OBJC_EXCHANGE_CLASS_METHOD_SAFE(const char* orgClassStr,
                               const char* orgSelector,
                               const char* newClassStr,
                               const char* newSelector);
#ifdef __cplusplus

#endif

CHookUtils.mm


#import "CHookUtils.h"
#import <objc/runtime.h>
bool OBJC_EXCHANGE_METHOD_SAFE(const char* orgClassStr,
                               const char* orgSelector,
                               const char* newClassStr,
                               const char* newSelector)

    Class orgClass = objc_getClass(orgClassStr);
    Class newClass = objc_getClass(newClassStr);
    if (!orgClass || !newClass) 
      return false;
    
    SEL orgMethod = sel_registerName(orgSelector);
    SEL newMethod = sel_registerName(newSelector);
    //class_getClassMethod
    Method newMethodIns = class_getInstanceMethod(newClass, newMethod);
    /*在旧的类添加类新方法后,那么只需要交换原有的方法即可*/
    Method orgMethodIns = class_getInstanceMethod(orgClass, orgMethod);
    /*防止方法不存在*/
    if (!orgMethodIns || !newMethodIns) 
      return false;
    
    /*在已有类添加新方法*/
    if (newClass != orgClass) 
      IMP newMethodIMP = method_getImplementation(newMethodIns);
      const char* newMethodStr = method_getTypeEncoding(newMethodIns);
      class_addMethod(orgClass, newMethod, newMethodIMP, newMethodStr);
      newMethodIns = class_getInstanceMethod(orgClass, newMethod);
    
    method_exchangeImplementations(orgMethodIns, newMethodIns);
    return true;



bool OBJC_EXCHANGE_CLASS_METHOD_SAFE(const char* orgClassStr,
                               const char* orgSelector,
                               const char* newClassStr,
                               const char* newSelector)

    Class orgClass = objc_getClass(orgClassStr);
    Class newClass = objc_getClass(newClassStr);
    if (!orgClass || !newClass) 
      return false;
    
    SEL orgMethod = sel_registerName(orgSelector);
    SEL newMethod = sel_registerName(newSelector);
    //class_getClassMethod
    Method newMethodIns = class_getClassMethod(newClass, newMethod);
    /*在旧的类添加类新方法后,那么只需要交换原有的方法即可*/
    Method orgMethodIns = class_getClassMethod(orgClass, orgMethod);
    /*防止方法不存在*/
    if (!orgMethodIns || !newMethodIns) 
      return false;
    
    /*在已有类添加新方法*/
    if (newClass != orgClass) 
      IMP newMethodIMP = method_getImplementation(newMethodIns);
      const char* newMethodStr = method_getTypeEncoding(newMethodIns);
      class_addMethod(orgClass, newMethod, newMethodIMP, newMethodStr);
      newMethodIns = class_getInstanceMethod(orgClass, newMethod);
    
    method_exchangeImplementations(orgMethodIns, newMethodIns);
    return true;

方法交换实现全局点击拦截

最后,博主在学习方法交换的过程中,发现用它是可以来做一些事情的,比如全局的点击统计。通过俺的研究,以及对iOS中事件传递事件的学习(这个以后专门写一篇分享),iOS中的事件主要会收拢在UIResponder和UIControl中,所以分别去hook了这两个类的touchesBegan方法和beginTrackingWithTouch方法,实现了对点击事件的收集,代码就不贴了比较多,感兴趣的同学可以把本文的源码都下载下来跑一跑,不需要积分哟。

源码地址

以上是关于iOS——面向AOP编程(方法交换)的主要内容,如果未能解决你的问题,请参考以下文章

JAVAEE框架整合技术之Spring02-AOP面向切面编程技术

JAVAEE框架整合技术之Spring02-AOP面向切面编程技术

Spring aop

spring_AOP编程

PHP AOP面向切面编程

iOS数据埋点统计方案(附Demo): 运行时Method Swizzling机制与AOP编程(面向切面编程)