Question

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

@interface A : NSObject

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

@end

类_B是A的私有子类：

@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时会发出什么以及如何更改此发出的代码？

Answer 1

差异很小，但很重要。编译器正在发出函数调用，而不是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的超类，这应该导致您的方法调用按预期执行。

ISA调高并调用`super`

1 个答案: