9

我在搞乱方法调配，想在执行method_exchangeImplementations. 我为此设置了两个项目。

第一个项目是应用程序的主项目。该项目包括应用程序的所有逻辑。请注意，originalMethodName当视图加载时调用。

@implementation ViewController

- (void)originalMethodName 
{
    NSLog(@"REAL %s", __func__);
}

- (void)viewDidLoad {
    [super viewDidLoad];

    NSLog(@"REAL %s", __func__);

    [self originalMethodName];
}

@end

第二个项目仅包含 swizzling 的代码。我有一个方法swizzle_originalMethodName，其中包含我想注入到主应用程序中的代码，originalMethodName并调用该函数。

@implementation swizzle_ViewController

- (void)swizzle_originalMethodName
{
    NSLog(@"FAKE %s", __func__);
}

__attribute__((constructor)) static void initializer(void)
{
    NSLog(@"FAKE %s", __func__);

    Class c1 = objc_getClass("ViewController");
    Class c2 = [swizzle_ViewController class];
    Method m1 = class_getInstanceMethod(c1, @selector(originalMethodName));
    Method m2 = class_getInstanceMethod(c2, @selector(swizzle_originalMethodName));
    method_exchangeImplementations(m1, m2);
}

@end

调酒器工作得很好（如下面的输出所示），但现在我希望能够originalMethodName从swizzle_originalMethodName

2016-08-17 14:18:51.765 testMacOS[7295:1297055] FAKE initializer
2016-08-17 14:18:51.822 testMacOS[7295:1297055] REAL -[ViewController viewDidLoad]
2016-08-17 14:18:51.822 testMacOS[7295:1297055] FAKE -[swizzle_ViewController swizzle_originalMethodName]

我曾尝试使用NSInvocation但没有任何运气。任何想法我做错了什么？

Class c1 = objc_getClass("ViewController");
Method m1 = class_getInstanceMethod(c1, @selector(originalMethodName));
NSMethodSignature *methodSignature = [NSMethodSignature signatureWithObjCTypes:method_getTypeEncoding(    m1)];
NSInvocation *originalInvocation = [NSInvocation invocationWithMethodSignature:methodSignature];
[originalInvocation invoke];
4

2 回答 2

11

如果您在类层次结构中进行调配，例如，您有一个子类将其祖先方法之一与自己的方法调配，那么您只需让调入方法显然调用自身——该调用实际上会调用调出方法因为方法已经交换了。在您的情况下，您将拥有：

- (void)swizzle_originalMethodName
{
   NSLog(@"FAKE %s", __func__);
   [self swizzle_originalMethodName]; // call original
}

这在您的情况下不起作用，因为您是跨类 swizzling，因此self不要使用 swizzled-out 方法引用该类。而且你没有 swizzling 类的实例，你可以调用 swizzled-out 方法......

这是解决此问题的一种简单方法，您的 swizzle-in 方法需要能够调用原始实现，您可以在设置 swizzling 时得到它。

在 Objective-C 中，方法由一个函数实现，该函数的前两个参数是调用该方法的对象引用，选择器和其余参数是该方法的参数。例如NSString方法：

- (NSRange)rangeOfString:(NSString *)aString

由类似的函数实现：

NSRange rangeOfStringImp(NSString *self, SEL cmd, NSString *aString)

您可以使用 获取指向此实现函数的函数指针method_getImplementation。

对于您的代码，首先在您swizzle_ViewController为您正在调配的方法的实现函数声明一个类型，以及一个用于存储函数指针的全局变量：

typedef void (*OriginalImpType)(id self, SEL selector);
static OriginalImpType originalImp;

现在在您的initializer方法中，您需要保存方法实现，您可以通过添加显示的行来做到这一点：

Method m1 = class_getInstanceMethod(c1, @selector(originalMethodName));
originalImp = (OriginalImpType)method_getImplementation(m1); // save the IMP of originalMethodName

最后让你的 swizzled 方法调用保存的实现：

- (void)swizzle_originalMethodName
{
   NSLog(@"FAKE %s", __func__);
   originalImp(self, @selector(originalMethodName)); // call the original IMP with the correct self & selector
}

可选：上述工作正常，但是它比要求的多一点——方法实现都是交换的，一个存储在全局变量中，你真正需要做的就是保存原始实现，m1然后将其实现设置为那个m2。您可以通过将调用替换为method_exchangeImplementations：

method_setImplementation(m1, method_getImplementation(m2));

打字要多一点，但对于实际需要做的事情要清楚一些。

高温高压

于 2016-08-18T00:48:21.480 回答
2

调用原始实现有一个稍微简单的选项，不需要您直接存储方法实现。当您交换方法的实现时，原始实现将存储在 swizzler 类中。class_getMethodImplementation您可以使用该函数获取混合后的实现。这是一个游乐场示例：

import Cocoa

let fooSelector = Selector("fooWithArg:")
let swizzledFooSelector = Selector("swizzled_fooWithArg:")

class A: NSObject {
    @objc dynamic func foo(arg: String) {
        print("Foo \(arg) in A")
    }
}

class B: NSObject {
    private typealias FooFunc = @convention(c) (AnyObject, Selector, String) -> Void
    @objc func swizzled_foo(arg: String) {
        print("Swizzled_foo \(arg) in B")

        unsafeBitCast(
            class_getMethodImplementation(B.self, swizzledFooSelector),
            to: FooFunc.self
        )(self, fooSelector, arg)
    }
}

method_exchangeImplementations(
    class_getInstanceMethod(A.self, fooSelector)!,
    class_getInstanceMethod(B.self, swizzledFooSelector)!
)

A().foo(arg: "bar")

调用 swizzled 的实现

于 2020-04-30T12:44:54.777 回答