iOS黑魔法之Method Swizzling

最新推荐文章于 2024-04-29 20:34:59 发布

原创最新推荐文章于 2024-04-29 20:34:59 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

iOS 同时被 2 个专栏收录

26 篇文章

订阅专栏

iOS专题入门

10 篇文章

订阅专栏

本文详细介绍了Objective-C中的Method Swizzling技术，这是一种基于Runtime的方法替换技术，允许开发者将现有方法替换成自定义方法，实现方法Hook的目的。文章通过具体实例展示了如何在UIViewController中替换viewWillAppear方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

此黑魔法本应属于OC，它是基于Runtime实现的偷天换日大法。

那么什么是Method Swizzling呢？从字面意思来看叫方法协调，大概就是这个意思了。使用此魔法可以替换现有方法为自定义方法，来个偷天换日，偷梁换柱。

使用方法很简单，代码基本为以下框架。但其具有较强的魔力，这是一个方法hook啊。

/**
 *  Method Swizzling
 *  黑魔法之偷天换日
 */
#import "UIViewController+Extension.h"
#import <objc/runtime.h>

@implementation UIViewController (Extension)

// 为了保证一定执行 把代码放到+ (void)load;里
+ (void)load {
    [super load];
    // 线程安全 只执行一次
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        // 获取class
        Class class = [self class];
//        Class class = object_getClass((id)self);
        // 封装selector
        SEL originalSelector = @selector(viewWillAppear:);
        SEL swizzledSelector = @selector(lw_viewWillAppear:);
        // 封装方法
        Method originalMethod = class_getInstanceMethod(class, originalSelector);
        Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
        // 添加方法
        BOOL methodDidAdd = class_addMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
        if(methodDidAdd) {
            // 替换方法
            class_replaceMethod(class, originalSelector, method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
        }else {
            // 交换方法
            method_exchangeImplementations(originalMethod, swizzledMethod);
        }
    });
}

#pragma mark - 自定义的魔法方法
- (void)lw_viewWillAppear:(BOOL)animated {
    [self lw_viewWillAppear:animated];
    NSLog(@"BLack Magic");
}

注意：

1 + (void)load;

2 dispatch_once;

3 [self lw_viewWillAppear:animated]; // 不回死循环若为[self viewWillAppear:animated];死循环