美文网首页
Aspects的源码分析问题点

Aspects的源码分析问题点

作者: Johnny_Wu | 来源:发表于2018-08-16 12:01 被阅读0次

    最近又看了遍Aspects源码,可以说每一次细看都可以收获到一些东西吧!这次解决了我之前的几个疑惑,具体分析下吧!

    一、之前说过,通过创建子类来实现,那么是怎样让方法调用定位到子类去呢?
    我们知道有个isa指针,就是控制去哪个类寻找IMP,所以最终是通过改变isa指针来实现。我们重新分析下Aspects创建子类的方法:

    static Class aspect_hookClass(NSObject *self, NSError **error) {
        NSCParameterAssert(self);
        Class statedClass = self.class;
        Class baseClass = object_getClass(self);
        NSString *className = NSStringFromClass(baseClass);
    
        // Already subclassed hook forwardInvocation这个方法
        if ([className hasSuffix:AspectsSubclassSuffix]) {
            return baseClass;
    
            // We swizzle a class object, not a single object.
        }else if (class_isMetaClass(baseClass)) {
            return aspect_swizzleClassInPlace((Class)self);
            // Probably a KVO'ed class. Swizzle in place. Also swizzle meta classes in place.
            //如果不等,就证明是self是类,不是对象
        }else if (statedClass != baseClass) {
            return aspect_swizzleClassInPlace(baseClass);
        }
    
        // Default case. Create dynamic subclass.
        const char *subclassName = [className stringByAppendingString:AspectsSubclassSuffix].UTF8String;
        Class subclass = objc_getClass(subclassName);
    
        //创建子类
        if (subclass == nil) {
            //继承与类baseClass的类
            subclass = objc_allocateClassPair(baseClass, subclassName, 0);
            if (subclass == nil) {
                NSString *errrorDesc = [NSString stringWithFormat:@"objc_allocateClassPair failed to allocate class %s.", subclassName];
                AspectError(AspectErrorFailedToAllocateClassPair, errrorDesc);
                return nil;
            }
    
            //hook forwardInvocation方法
            aspect_swizzleForwardInvocation(subclass);
            aspect_hookedGetClass(subclass, statedClass);
            aspect_hookedGetClass(object_getClass(subclass), statedClass);
            objc_registerClassPair(subclass);
        }
        //与object_getClass对应,object_getClass是获得isa指针,那么object_setClass是重定向isa执行subclass
        object_setClass(self, subclass);
        return subclass;
    }
    

    前面的判断是,如果是传入的是类,那么就直接进行hook,如果传入的是对象,就创建子类,然后hook子类的方法。也许这里创建子类也不是必要的吧,因为我直接调用:

    return aspect_swizzleClassInPlace(baseClass);
    

    好像也是可以的。
    如果创建了子类,那么就得考虑重定向问题。我们在代码最后发现:

    object_setClass(self, subclass);
    

    与object_getClass对应,object_getClass是获得isa指针,那么object_setClass是重定向isa执行subclass。所以这就实现了重定向。

    aspect_hookedGetClass(subclass, statedClass);
    aspect_hookedGetClass(object_getClass(subclass), statedClass);
    

    因为返回的是子类了,如果调用subclass.class或object_getClass(subclass).class,又让它返回的是原来的类。可能这样做是为了避免一些特殊的情况吧。

    二、为什么没实现methodSignatureForSelector方法,最后也可以调用forwardInvocation
    这里面,涉及到:

    class_replaceMethod(klass, selector, aspect_getMsgForwardIMP(self, selector), typeEncoding);
    

    我之前认为aspect_getMsgForwardIMP是空实现,没实现后,就会最后走到forwardInvocation这个方法。但我又发现源码好像只实现了forwardInvocation方法,没实现methodSignatureForSelector返回签名的方法,按我的理解,没有签名的话是会崩溃的啊。
    所以,答案是,我对aspect_getMsgForwardIMP理解有误。下面我通过实验验证一下。

    @interface MyModel : NSObject
    
    - (void)gainMoney:(NSString *)name;
    
    @end
    @implementation MyModel
    
    - (void)gainMoney:(NSString *)name
    {
        NSLog(@"MyModel money");
    }
    
    + (BOOL)resolveInstanceMethod:(SEL)sel      //实现没实现的方法,有实现方法的话就不会走下面的方法了
    {
        NSLog(@".........1...........");
        return [super resolveInstanceMethod:sel];
    }
    //消息转发
    - (id)forwardingTargetForSelector:(SEL)aSelector        //没有重新实现方法(要是正确的实现方法),就会调用这个
    {
        NSLog(@".........2...........");
        return nil;
    }
    
    //另外一种消息转发,手动生成方法签名
    - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector //没正确指向方法,就调用这个
    {
        NSLog(@".........3...........");
        return [super methodSignatureForSelector:aSelector];
    }
    //生成签名成功后,就会调用下面的方法 NSInvocation *invocation是通过NSMethodSignature得到的
    - (void)forwardInvocation:(NSInvocation *)anInvocation
    {
        NSLog(@".........4...........");
        NSString *str = @"";
        //参数
        [anInvocation getArgument:&str atIndex:2];
        NSLog(@"第一个参数:%@",str);
    }
    @end
    

    上面定义了 MyModel类,下面是测试代码,使用的Aspects的源码部分,用于把gainMoney指向_objc_msgForward。

    - (void)testMsgChange
    {
        MyModel *model = [[MyModel alloc] init];
        //获得方法
        Method targetMethod = class_getInstanceMethod(model.class, @selector(gainMoney:));
        //获取原来方法参数
        const char *typeEncoding = method_getTypeEncoding(targetMethod);
        //重新指向方法,把方法直接指向forwardInvocation
        class_replaceMethod(model.class, @selector(gainMoney:), _objc_msgForward, typeEncoding);
        [model gainMoney:@"123"];
    }
    

    结果如下:
    2018-08-16 11:50:30.997663+0800 SummaryTest[3068:2232826] .........2...........
    2018-08-16 11:50:30.997844+0800 SummaryTest[3068:2232826] .........3...........
    2018-08-16 11:50:30.997992+0800 SummaryTest[3068:2232826] .........4...........
    2018-08-16 11:50:30.998127+0800 SummaryTest[3068:2232826] 第一个参数:123

    可以看到,原方法指向_objc_msgForward,不是指向空,最后是可以走到forwardInvocation函数的。

    相关文章

      网友评论

          本文标题:Aspects的源码分析问题点

          本文链接:https://www.haomeiwen.com/subject/lesnbftx.html