Method Swizzle黑魔法，修改 ios 系统类库方法(转载)

一般来说，系统提供的方法已经足够开发了，但是有的时候有些需求用普通方法不好做。

如：在所有的viewcontroll 的viewwillappear：方法之前打个log

你可能会这么做：

1. 建一个uiviewcontroll 父类，重写viewwillappear方法，调用super viewwillappear 方法之前加上log

2. 所有新建的uiviewcontroller 继承第一步生成的

确实你是完成这样的功能，可是你做了那么多的修改，基本每个uiviewcontroller都去修改了父类，这种方法太过于笨重了

 

本文提供了简单地方法即可实现

我的理解中，object-c 的类调用方法是根据三个元素来定义的。 

1. 方法，代表类定义中一个方法类型（typedef struct objc_method *Method）

2. SEL 选择器（typedef struct objc_selector *SEL），一个方法在运行时的名字，常见的有 [self performSelector:@selector(somemethod:) withObject:nil afterDelay:0.5]; @selector(somemethod:)作为方法的入口

3. 方法的实现入口（typedef id (*IMP)(id, SEL, …)）

这三个元素确定了具体调用哪一个函数

 

直接看代码

 

#import "UIViewController+Tracking.h"
#import <objc/runtime.h>

@implementation UIViewController (Tracking)

+ (void)load {
    NSString *className = NSStringFromClass(self.class);
    NSLog(@"classname %@", className);
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];

        // When swizzling a class method, use the following:
        // Class class = object_getClass((id)self);

        SEL originalSelector = @selector(viewWillAppear:);
        SEL swizzledSelector = @selector(xxx_viewWillAppear:);

        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

        BOOL didAddMethod =
        class_addMethod(class,
                        originalSelector,
                        method_getImplementation(swizzledMethod),
                        method_getTypeEncoding(swizzledMethod));

        if (didAddMethod) {
            class_replaceMethod(class,
                                swizzledSelector,
                                method_getImplementation(originalMethod),
                                method_getTypeEncoding(originalMethod));
        } else {
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
    });
}

 

我们category重写了NSObject的 load 方法oc提供了objc/runtime.h类让我们获取这些东西，同时还提供了对类方法操作的函数

我们想的是，直接用一个方法替换掉系统的方法，然后把一些自定义的动作加到方法中

我们只想运行一次就够了，所以使用了 dispatch_once(&onceToken, ^{ …… }
接下来给类添加了新方法

把新方法和系统方法替换

 

 

#pragma mark - Method Swizzling

- (void)xxx_viewWillAppear:(BOOL)animated {
    NSLog(@"viewWillAppear: %@", self);
    [self xxx_viewWillAppear:animated];
}

但是新方法实现的时候，调用的是 [self xxx_viewwillAppear:animated]; 可能你会疑惑

 

这是因为我们在上面已经用xxx_viewwillAppear 和 viewwillAppear 互换了。所以实际上执行的是系统的viewwillAppear

这个时候可能你又有疑问了，为什么实现是- (void)xxx_viewWillAppear:(BOOL)animated{} 这样的

这是因为 SEL swizzledSelector = @selector(xxx_viewWillAppear:); 拿的就是我们新写的方法。

可以结合这篇博客看，配图很容易懂

http://blog.csdn.net/yiyaaixuexi/article/details/9374411

 

以及这篇对SEL讲的比较清楚

http://blog.csdn.net/fengsh998/article/details/8612969

代码下载地址

https://github.com/holysin/Method_swizzle

1  Method nameMethod = class_getInstanceMethod(Class class,SEL name);//得到实例方法

参数:class类名  name 方法名

理解:通过类名和方法名来得到方法(c函数)

 

2 method_getImplementation(Method method)

参数:method 方法

理解:通过方法来得到它的实现

 

 

3 BOOL class_addMethod(Class cls, SEL name, IMP imp, const char *types)

理解:将所给的name(方法名)的方法实现,被imp(方法的实现)代替

cls：被添加方法的类

name：可以理解为方法名，这个貌似随便起名，

imp：实现这个方法的函数 

types：一个定义该函数返回值类型和参数类型的字符串，这个具体会在后面讲

 

4 class_replaceMethod(Class class,SEL  B, method_getImplementation(Method AMethod), method_getTypeEncoding(Method AMethod));

参数:class :方法所属的类 B:将被替换其实现的方法名    AMethod:由A生成的方法(c函数)

理解:用A的方法实现来代替B的方法实现

5 method_exchangeImplementations(SEL A, SEL B);

理解:交换两个方法的实现

void swizzle_method(Class class,SEL originalSelector,SEL swizzledSelector) {

    Method originalMethod = class_getInstanceMethod(class, originalSelector);//得到实例方法

    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);

    BOOL didSwizzleMethod = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));

    if (didSwizzleMethod) {

        class_replaceMethod(class, swizzledSelector, method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));

    } else {

        method_exchangeImplementations(originalMethod, swizzledMethod);

    }

}