ISA调高并调用`super`

时间:2015-02-03 12:29:50

标签: objective-c superclass objective-c-runtime swizzling isa-swizzling

假设以下类层次结构。类A已公开声明:

@interface A : NSObject

+ (A)createInstance;
- (void)a;

@end

_BA的私有子类:

@interface _B : A

- (void)a;
- (void)b;

@end

假设只应使用工厂方法A创建类createInstance的对象,该方法创建并返回_B的实例。

我希望在每个实例的基础上增强A实例的功能。所以我决定做一些ISA调整来实现:

@interface ExtA : A

- (void)a;

@end

@implementation ExtA

- (void)a
{
    NSLog("ExtA_a");
    [super a];
}

@end

我在NSObject类别上使用以下方法进行ISA调配(此处显示的天真实现):

- (void)changeToSubclass:(Class)cls prefix:(NSString*)prefix suffix:(NSString*)suffix
{
    NSString* className = [NSString stringWithFormat:@"%@%@%@", prefix ? prefix : @"", NSStringFromClass(object_getClass(self)), suffix ? suffix : @""];

    if([className isEqualToString:NSStringFromClass(object_getClass(self))])
    {
        className = [NSString stringWithFormat:@"%@(%@)", NSStringFromClass(object_getClass(self)), NSStringFromClass(cls)];
    }

    Class newSubclass = objc_getClass(className.UTF8String);

    if(newSubclass == nil)
    {
        newSubclass = objc_allocateClassPair(object_getClass(self), className.UTF8String, 0);
        objc_registerClassPair(newSubclass);

        unsigned int listCount = 0;
        Method *list = class_copyMethodList(cls, &listCount);

        for(int i = 0; i < listCount; i++)
        {
            class_addMethod(newSubclass, method_getName(list[i]), method_getImplementation(list[i]), method_getTypeEncoding(list[i]));
        }
        free(list);

        listCount = 0;
        list = class_copyMethodList(objc_getMetaClass(class_getName(cls)), &listCount);

        for(int i = 0; i < listCount; i++)
        {
            class_addMethod(objc_getMetaClass(class_getName(newSubclass)), method_getName(list[i]), method_getImplementation(list[i]), method_getTypeEncoding(list[i]));
        }
        free(list);
    }

    object_setClass(self, newSubclass);
}

一切似乎都有效,但我注意到[super a];的行为并不像预期的那样,实际上调用-[A a]的实现,如果运行时的超类实际上是_B

使用以下代码替换对super的调用有效,但是很难看,并且需要开发人员的知识和工作:

struct objc_super superInfo = {
    self,
    [self superclass]
};
objc_msgSendSuper(&superInfo, @selector(a));

编译器在调用super时会发出什么以及如何更改此发出的代码?

1 个答案:

答案 0 :(得分:2)

差异很小,但很重要。编译器正在发出函数调用,而不是objc_msgSendSuper,而是objc_msgSendSuper2

你可能会问,有什么区别?它很小,但很重要。

来自apple's open source

/********************************************************************
 *
 * id objc_msgSendSuper(struct objc_super *super, SEL _cmd,...);
 *
 * struct objc_super {
 *      id  receiver;
 *      Class   class;
 * };
 ********************************************************************/

    ENTRY   _objc_msgSendSuper
    MESSENGER_START

// search the cache (objc_super in %a1)
    movq    class(%a1), %r11    // class = objc_super->class
    CacheLookup SUPER       // calls IMP on success

/* Snipped code for brevity */

/********************************************************************
 * id objc_msgSendSuper2
 ********************************************************************/

    ENTRY _objc_msgSendSuper2
    MESSENGER_START

    // objc_super->class is superclass of class to search

// search the cache (objc_super in %a1)
    movq    class(%a1), %r11    // cls = objc_super->class
    movq    8(%r11), %r11       // cls = class->superclass
    CacheLookup SUPER2      // calls IMP on success

对于那些不熟悉x86_64程序集的读者,重要的代码行在这里:

movq    8(%r11), %r11       // cls = class->superclass

你可能会问,这是做什么的?它非常简单 - 而不是调用者将超类传递给搜索,而objc_msgSend实现就是这样做。

然而,这一重要区别会导致一个至关重要的问题 - 在进行super调用时,会调用[self class]。相反,它使用当前实现的类,当然,ExtA

因此,唯一的方法是“修复”。这是在运行时更改ExtA的超类,这应该导致您的方法调用按预期执行。