Method Swizzling(iOS的hook机制)

为了安全起见，要为NSUserDefaults中的数据加密。但是在每个调用NSUserDefaults读写方法的位置，加入加密解密方法也太麻烦。所以想重写NSUserDefaults的读写方法，把加密解密方法内嵌其中，这样修改代码量会大大减小。

先贴上代码

#import <Foundation/Foundation.h>

@interface NSUserDefaults (Swizzle)

-(void)swizzleSetObject:(nullable id) value forKey:(nullable NSString *)defaultName;
-(nullable id)swizzleObjectForKey:(NSString *)defaultName;

@end

#import "NSUserDefaults+Swizzle.h"
#import "WFTTripleDES.h"
@implementation NSUserDefaults (Swizzle)

-(void)swizzleSetObject:(nullable id) value forKey:(nullable NSString *)defaultName
{
    if ([value isKindOfClass:[NSString class]]) {
        //加密后写入
        [self swizzleSetObject:[WFTTripleDES TripleDES:value encryptOrDecrypt:kCCEncrypt key:[NSString stringWithFormat:@"2016012031009%@",defaultName] ] forKey:defaultName];
    }else{
        [self swizzleSetObject:value forKey:defaultName];
    }
}

-(nullable id)swizzleObjectForKey:(NSString *)defaultName;
{
    //如果键值为空直接返回
    if (!defaultName) {
        return @"";
    }
    
    id  getObject = [self swizzleObjectForKey:defaultName];
    if ([getObject isKindOfClass:[NSString class]]) {
        //把结果解密
        return [WFTTripleDES TripleDES:getObject encryptOrDecrypt:kCCDecrypt key:[NSString stringWithFormat:@"2016012031009%@",defaultName]];
    }else{
        return getObject;
    }
}
@end

然后交换自定义方法和系统方法

   Method originalMethod = class_getInstanceMethod([NSUserDefaults class],  @selector(setObject:forKey:));
    Method swizzledMethod = class_getInstanceMethod([NSUserDefaults class], @selector(swizzleSetObject:forKey:));
    method_exchangeImplementations(originalMethod, swizzledMethod);

Method Swizzling 原理

在Objective-C中调用一个方法，其实是向一个对象发送消息，查找消息的唯一依据是selector的名字。利用Objective-C的动态特性，可以实现在运行时偷换selector对应的方法实现，达到给方法挂钩的目的。

每个类都有一个方法列表，存放着selector的名字和方法实现的映射关系。IMP有点类似函数指针，指向具体的Method实现。

我们可以利用 method_exchangeImplementations 来交换2个方法中的IMP，

上面的代码实现

然后调用method_exchangeImplementations方法交换后变成

所以调用setObject:forKey:方法时，其实调用的是swizzleSetObject:forKey:的实现。而调用swizzleSetObject:forKey:方法时，调用的系统的setObject:forKey:的实现。

如果在swizzleSetObject:forKey:实现方法中，调用setObject:forKey:方法,其实会回调swizzleSetObject:forKey:。