如何实现方法调配?

时间:2011-03-20 21:23:47

标签: objective-c cocoa methods swizzling simbl

我正在尝试使用SIMBL修改程序的行为(我没有它的源代码)。我使用了类转储,发现我需要覆盖一个实例方法

此方法位于名为controller的类中。我需要做的就是获得参数arg1,就是这样。也许NSLog它或发布通知... 我在objective-c中读到了关于方法调配的内容,但我该如何使用呢?我需要引用我没有的课程MessageController。

谢谢!

2 个答案:

答案 0 :(得分:33)

我猜你在做完NSLog之后需要调用原来的实现;如果没有,您可以只使用类上的类别来覆盖该方法。

要调整方法,首先需要更换方法。我通常把这样的东西放在目标类的一个类别中:

- (void)replacementReceiveMessage:(const struct BInstantMessage *)arg1 {
    NSLog(@"arg1 is %@", arg1);
    [self replacementReceiveMessage:arg1];
}

这看起来会以递归方式调用自身,但它不会因为我们要交换周围的内容所以在调用ReceiveMessage:时调用replacementReceiveMessage:调用方法调用旧版本。

第二步是使用运行时函数来实际执行交换。使用类别的优点是您可以在类别中使用load来完成工作:

+ (void)load {
    SEL originalSelector = @selector(ReceiveMessage:);
    SEL overrideSelector = @selector(replacementReceiveMessage:);
    Method originalMethod = class_getInstanceMethod(self, originalSelector);
    Method overrideMethod = class_getInstanceMethod(self, overrideSelector);
    if (class_addMethod(self, originalSelector, method_getImplementation(overrideMethod), method_getTypeEncoding(overrideMethod))) {
            class_replaceMethod(self, overrideSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
    } else {
            method_exchangeImplementations(originalMethod, overrideMethod);
    }
}

有两种情况需要处理:

  • 如果我们调配的方法实际上是在超类中定义的,我们必须使用class_addMethod向目标类添加ReceiveMessage:的实现,我们使用替换实现来实现。{1}}。然后我们可以使用class_replaceMethodreplacementReceiveMessage:替换为超类的实现,因此我们的新版本将能够正确地调用旧版本。
  • 如果方法是在目标类中定义的,class_addMethod将失败,但我们可以使用method_exchangeImplementations来交换新版本和旧版本。

答案 1 :(得分:6)

jrswizzle处理它。不建议您自己动手,因为有很多细节要做。 (请参阅jrswizzle自述文件中记录以前实现失败的表格。)

假设你有这样的课程:

@interface Weh : NSObject
-(void)foo;
-(void)bar;
@end

@implementation Weh
-(void)foo {
    NSLog(@"Foo called");
}
-(void)bar {
    NSLog(@"Bar called");
    [self bar];
}
@end

你可以像这样使用它:

Weh *weh = Weh.new;
[weh foo];
[Weh jr_swizzleMethod:@selector(foo) withMethod:@selector(bar) error:nil];
[weh foo]; 

输出:

Foo called
Bar called
Foo called