iOS中几种hook代码的方法

1.  Method Swizzle

        iOS中Object-C代码的主要hook方式。其原理是利用OC的Runtime特性，动态改变SEL（方法编号）和IMP（方法实现）的对应关系，达到OC方法调用流程改变的目的。

        比如， 我们可以通过hook字典NSMutableDictionary的 setObject:forKey和removeObjectForKey:方法，帮助其实现空安全。

#import "NSMutableDictionary+NullSaf.h"
#import<objc/runtime.h>

@implementation NSMutableDictionary (Safe)
+ (void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        id obj = [[self alloc]init];
        [obj swizzeMethod:@selector(setObject:forKey:) withMethod:@selector(safe_setObject:forKey:)];
        [obj swizzeMethod:@selector(removeObjectForKey:) withMethod:@selector(safe_removeObjectForKey:)];
    });
}

- (void)swizzeMethod:(SEL)origSelector withMethod:(SEL)newSelector{
    Class class = [self class];
    Method originalMethod = class_getInstanceMethod(class, origSelector);
    Method swizzledMethod = class_getInstanceMethod(class, newSelector);
    BOOL didAddMethod = class_addMethod(class, origSelector,
method_getImplementation(swizzledMethod), method_getTypeEncoding(swizzledMethod));
    if (didAddMethod) {
        class_replaceMethod(class, newSelector,
method_getImplementation(originalMethod), method_getTypeEncoding(originalMethod));
    }
    else{
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}

- (void)safe_setObject:(id)value forKey:(NSString* )key{
    if (value) {
        [self safe_setObject:value forKey:key];
    }else{
        NSLog(@"[NSMutableDictionary setObject: forKey:%@]值不能为空;",key);
    }
}

- (void)safe_removeObjectForKey:(NSString *)key{
    if ([self objectForKey:key]) {
        [self safe_removeObjectForKey:key];
    }else{
        NSLog(@"[NSMutableDictionary setObject: forKey:%@]值不能为空;",key);
    }
}


@end

2. inline hook: dobby

       dobby是一个开源的第三方框架，主要用于C、C++等静态代码的hook。其原理是，直接修改目标函数的头部代码（修改的是内存中MachO的代码段__TEXT），让它跳转到我们自定义的函数里面执行我们的代码，从而达到Hook的目的。

       具体而言， hook过程主要包含以下三步：

     （1）将原函数的前 N 个字节搬运到 Hook 函数的前 N 个字节；

     （2）然后将原函数的前 N 个字节填充为跳转到 Hook 函数的指令；

     （3）在 Hook 函数末尾几个字节填充跳转回原函数+N 的跳转指令，使得原来的代码继续执行；

        dobby的使用也非常简单，直接调用DobbyHook方法即可。 如下所示：

- (void)viewDidLoad {
    [super viewDidLoad];
    DobbyHook(sum, mySum, (void *)&sum_p);
}
int sum(int a, int b) {
    return a+b;
}
//函数指针用于保留原来的执行
static int (*sum_p)(int a,int b);
//新函数地址
int mySum(int a,int b) {
    NSLog(@"HOOK success,原来的结果是 %d",sum_p(a,b));
    return a-b;
}
- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
    NSLog(@"output result %d",sum(10, 20));
}

        点击屏幕， 输出的结果是： output result -10

3. FaceBook的fishHook

        动态库mach-o中存储的符号表，在运行时才进行从符号到内存的绑定，fishHook的原理即是，利用MachO文件加载原理，通过修改 懒加载和非懒加载 两个表的指针将原方法的符号，绑定为hook方法的符号地址，即可完成hook。具体来说，就是通过使用  rebind_symbol方法，修改两个符号对应的地址，进而实现符号的重新绑定。

       

        由于符号绑定主要是针对当前镜像外部的函数， 因此fishHook使用上有一定局限性。即只有系统动态库中的函数，或者其他非当前镜像中的函数，才可以通过重新进行符号绑定，实现对其进行hook。

       

#import "ViewController.h"
#import "fishhook.h"
 
@implementation ViewController
- (void)viewDidLoad {
    [super viewDidLoad];

    //使用rebinding 结构体
    struct rebinding nslog;
    nslog.name = "NSLog";//要hook的函数名称
    nslog.replacement = myNSLog;//这里是函数的指针，也就是函数名称
    nslog.replaced = &sys_nslog;

    struct rebinding rebs[1] = {nslog};
    rebind_symbols(rebs, 1);
}

//NSLog函数指针
static void(*sys_nslog)(NSString * format,...);

//定义一个新函数
void myNSLog(NSString *format,...){
    format = [format stringByAppendingString:@"~~~~~hook 到了!"];
    sys_nslog(format);
}

-(void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
    NSLog(@"hello");
}

        代码的执行结果：hello~~~~~hook 到了!