iOS面试题目——hook block（2）

// 题目：实现下面的函数，将 block 的实现修改成打印所有入参，并调用原始实现
//
// 例如：
//  void(^block)(int a, NSString *b) = ^(int a, NSString *b){
//      NSLog(@"block invoke");
//  }
//  HookBlockToPrintArguments(block);
//  block(123,@"aaa");
//  这里输出 "123,aaa" 和 "block invoke"

分析：这个题目其实和题目一的本质是一样的，都是替换block的实现（即Hook Block），不过，相比较于题目一，这个题的侧重点在于：1、打印所有入参；2、调用原实现。针对这两个问题，我们逐一解析。

1、打印所有入参

对于已知参数个数和参数类型的block，要实现这个，其实并不难，只需要我们再声明替换函数的时候和block的参数对齐即可：

//这里要注意的是，第一个参数必须声明为block本身。
//针对	void(^block)(int a, NSString *b) ,我们可以将函数声明为如下形式：
void replace_bloke2(id block, int a, NSString *b);

2、调用原实现

//声明一个函数指针，用来存储invoke的值
void(*origin_blockInvoke2)(id block,int a,NSString *b);

void replace_bloke2(id block, int a, NSString *b) {
    NSLog(@"%d,%@",a,b);
    origin_blockInvoke2(block,a,b);
}

void HookBlockToPrintArguments(id block){

//    解析 block 为 struct Block_layout 结构体
    struct Block_layout *layout = (__bridge struct Block_layout *)block;
//    修改内存属性
    vm_address_t invoke_addr = (vm_address_t)&layout->invoke;
    vm_size_t vmsize = 0;
    mach_port_t object = 0;
    vm_region_basic_info_data_64_t info;
    mach_msg_type_number_t infoCnt = VM_REGION_BASIC_INFO_COUNT_64;
    kern_return_t ret = vm_region_64(mach_task_self(), &invoke_addr, &vmsize, VM_REGION_BASIC_INFO, (vm_region_info_t)&info, &infoCnt, &object);
if (ret != KERN_SUCCESS) {
        NSLog(@"获取失败");
return;
    }
    vm_prot_t protection = info.protection;
//    判断内存是否可写
if ((protection&VM_PROT_WRITE) == 0) {
//        修改内存属性 ===> 可写
        ret = vm_protect(mach_task_self(), invoke_addr, sizeof(invoke_addr), false, protection|VM_PROT_WRITE);
if (ret != KERN_SUCCESS) {
            NSLog(@"修改失败");
return;
        }
    }
//    保存原来的invoke
    origin_blockInvoke2 = (void *)layout->invoke;
    layout->invoke = (uintptr_t)replace_bloke2;
}