iOS Hook第三方静态库实践(1)

2019-07-22  本文已影响0人  zjam9333

在一次接入某个必须的第三方静态库,此库提供蓝牙连接某硬件和加密交互的功能,发现了一些逻辑问题,在连接的方法- (BOOL) ConnectDevice: (NSString *)DevName Timeout:(uint8_t) byTimeOut;只有设备名和超时时间两个参数。

如果附近有多个这种设备,我将无法连接具体某一个设备(例如信号最强的设备、某个特定序列号的设备等),而只能按照它内部的逻辑,不受控制的连接了。例如连接到了隔壁房间的设备,就很影响心情了。

此库公开的头文件

因为必须用它的连接方法才能正常使用,但又不好用。于是我想看看这个连接相关逻辑有没有hook的可能,改成可以连接指定的CBPeripheral对象。

那么开始做恶吧!

1. 新建简单项目,引入第三方库,打包获取二进制文件

为了更方便的分析过程,项目中最好只放入要分析的这个库,到后面就不会眼花。
为什么不直接使用这个库里面的二进制文件呢?
因为这样的话,下一步会发现有好多个.o文件,根本不知道应该看哪一个,而且每次都选择一个.o文件来打开会麻烦。

使用ipa文件里的二进制文件
2. 使用Hopper Disassembler分析

打开Hopper Disassembler(试用版或其他版本)

打开刚才的二进制文件

如果打开的是framework中的二进制文件,将会看见很多个.o文件要选择。

直接使用framework中的二进制文件会遇到麻烦

于是使用ipa文件里的二进制文件,点击下一步就好了,然后就会看见如下界面。尽可能等上方的进度条不再变化时才继续操作,否则有crash的可能。

某界面 直接找到连接相关方法

默认是显示成汇编代码,一般人都看不懂啊(我就是那种一般人)。那么点击上方的if(b)f(x):按钮可以查看比较直观的代码,但仍然比较奇怪,各种rxx变量、0xx地址等,不过已经能大致看懂了。

如图所示,它里面使用了一个BleManager的类的connectBtDevice:timeOut:去做这个具体的连接操作,其中r22arg2(也就是最开始的NSString *DevName)。那么直接进到BleManagerconnectBtDevice:timeOut:里面看看到底干了什么。

BleManager的connectBtDevice:timeOut:

大致可以看出,它内部用了一个CBCentralManager又做了一次搜索设备的过程,然后去连接,代理就是这个BleManager对象。(居然还用了一个runloop去卡线程控制超时!!!)

那么我们看看它怎么实现这个CBCentralManagerDelegate

先看看最重要的 centralManager:didDiscoverPeripheral:advertisementData:RSSI:,一般在这个方法中能获取搜索到的peripheral,再决定是否连接它。

BleManager 的centralManager:didDiscoverPeripheral:advertisementData:RSSI:

它在centralManager:didDiscoverPeripheral:advertisementData:RSSI:代理回调中,判断peripheral的名字是否正确,正确则连接。

那么大致弄明白了,修改BleManagerconnectBtDevice:timeOut:centralManager:didDiscoverPeripheral:advertisementData:RSSI:便可以实现自定义连接。而且connectBtDevice:timeOut:可传入id类型(即指定的peripheral)。

3. 新建Category,入侵原有方法

Objective-Chook一般是通过方法交换实现的。那么先实现通用的方法交换。

例如从网上摘抄的代码:

#import <objc/runtime.h>

@interface NSObject (SwizzlingMethod)

+ (void)swizzleSelector:(SEL)originalSelector withSelector:(SEL)swizzledSelector;

@end

@implementation NSObject (SwizzlingMethod)

+ (void)swizzleSelector:(SEL)originalSelector withSelector:(SEL)swizzledSelector {
    
    Class class = [self class];
    
    Method originalMethod = class_getInstanceMethod(class, originalSelector);
    Method swizzledMethod = class_getInstanceMethod(class, swizzledSelector);
    // 注意class_getInstanceMethod和class_getClassMethod的区别
    
    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);
    }
}

@end

因为要修改BleManagerconnectBtDevice:timeOut:centralManager:didDiscoverPeripheral:advertisementData:RSSI:,所以创建一个BleManager的分类,我想要实现连接信号最强的那一个(如果有多个这样的设备),那么新增几个变量如下。

static NSInteger _strongestRSSI;
static CBPeripheral *_nearestPeripheral;
static __weak CBCentralManager *_btIdRealCentralManager; 
// 此Manager和Peripheral可在centralManager:didDiscoverPeripheral:advertisementData:RSSI:方法回调中直接获得

因为编译器不知道有BleManager这个类,所以先声明。
那么整个BleManager的分类看起来大致就是这样:

#import <Foundation/Foundation.h>
#import <CoreBluetooth/CoreBluetooth.h>

static NSInteger _strongestRSSI;
static CBPeripheral *_nearestPeripheral;
static __weak CBCentralManager *_btIdRealCentralManager;

@interface BleManager : NSObject

@end

@implementation BleManager(ConnectBetter)

+ (void)load {
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        // 重写新的方法来交换BleManager的connectBtDevice:timeOut:和centralManager:didDiscoverPeripheral:advertisementData:RSSI:
        [self swizzleSelector:NSSelectorFromString(@"connectBtDevice:timeOut:") withSelector:@selector(myConnectBtDevice:timeOut:)];
        [self swizzleSelector:@selector(centralManager:didDiscoverPeripheral:advertisementData:RSSI:) withSelector:@selector(myCentralManager:didDiscoverPeripheral:advertisementData:RSSI:)];
    });
}

- (bool)myConnectBtDevice:(id)someDevice timeOut:(uint8_t)arg3 {
    // blablabla
}

- (void)myCentralManager:(CBCentralManager *)central didDiscoverPeripheral:(CBPeripheral *)peripheral advertisementData:(NSDictionary *)advertisementData RSSI:(NSNumber *)RSSI {
    // blablabla
}

@end

myConnectBtDevice:timeOut:myCentralManager:didDiscoverPeripheral:advertisementData:RSSI:的具体实现如下:

- (bool)myConnectBtDevice:(id)someDevice timeOut:(uint8_t)arg3 {
    NSString *deviceName = someDevice;
    BOOL isPeripheral = [someDevice isKindOfClass:[CBPeripheral class]];
    if (isPeripheral) {
        deviceName = [someDevice name];
    }
    _strongestRSSI = -1000;
    _nearestPeripheral = nil;
    __block BOOL missed = NO;
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(2 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        if (isPeripheral) {
            _nearestPeripheral = someDevice;
        }
        if (_nearestPeripheral && missed == NO) {
            [_btIdRealCentralManager connectPeripheral:_nearestPeripheral options:nil];
//            _nearestPeripheral = nil;
        }
    });
    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)((arg3 - 1) * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
        _nearestPeripheral = nil;
        missed = YES;
    });
    return [self myConnectBtDevice:deviceName timeOut:arg3];
}

- (void)myCentralManager:(CBCentralManager *)central didDiscoverPeripheral:(CBPeripheral *)peripheral advertisementData:(NSDictionary *)advertisementData RSSI:(NSNumber *)RSSI {
    _btIdRealCentralManager = central;
    NSData *facturerData = [advertisementData valueForKey:@"kCBAdvDataManufacturerData"];
    //    NSLog(@"%@", facturerData);
    if (facturerData.length > 6) {
        NSString *facturerStr = [[NSString alloc] initWithData:facturerData encoding:NSASCIIStringEncoding];
        if ([facturerStr hasPrefix:@"我想要的工厂信息"]) {
            NSLog(@"facture:%@, RSSI:%@", facturerStr, RSSI);
            NSInteger rssiValue = RSSI.integerValue;
            if (rssiValue > _strongestRSSI) {
                _strongestRSSI = rssiValue;
                _nearestPeripheral = peripheral;
                NSLog(@"check nearest:%@, RSSI:%@", facturerStr, RSSI);
            }
        }
    }
}
4. 最后的最后

请仔细测试多几次!确保不会crash!!确保功能符合预期!!!

上一篇下一篇

猜你喜欢

热点阅读