×

获取NSURLResponse的HTTPVersion

96
appfish
2017.01.14 21:22* 字数 672

一、需求背景

NSURLProtocol可以拦截UIWebView/WKWebView里的请求,公司的产品需要将拦劫到的HTTPS请求的URL的Host改成我们自己特定的Host,然后在请求头里带上原始域名给我们的节点,节点再以原始域名去访问源站。我们称之为HTTPS域名收敛。

域名被收敛之后面临着一个问题,就是网页里的html/css/js等文件被收敛之后,里面很多用相对地址编码的资源被WebView加载时域名也变成了我们那个特定域名,这样这些资源的原始域名就丢失了,导致WebView里很多资源加载失败。问题的解决方法是在NSURLProtocol里做域名收敛收到NSHTTPURLResponse的时候,将它的URL的Host还原成原来的域名。

二、修改NSHTTPURLResponse的URL

NSHTTPURLResponse的URL属性是只读的,也没有被KVC,我们无法简单的去直接修改它。所以只能根据原来的Response重新再构造出一个新的URL不一样的Response。NSHTTPURLResponse初始化方法如下

- (nullable instancetype)initWithURL:(NSURL *)url
                        statusCode:(NSInteger)statusCode
                         HTTPVersion:(nullable NSString *)HTTPVersion
                        headerFields:(nullable NSDictionary<NSString *, NSString *> *)headerFields;

构造NSHTTPURLResponse所需的其他参数都好说,唯独HTTPVersion这个参数我们无法直接得到,也就是没有从原来那个NSHTTPURLResponse获取HTTPVersion的Public API或者Private API。

三、获取NSURLResponse的HTTPVersion

查资料后知道NSURLResponse是基于_CFURLResponse这个结构体实现的

typedef struct _CFURLResponse {
    CFRuntimeBase _base;
    CFAbsoluteTime creationTime;
    CFURLRef url;
    CFStringRef mimeType;
    int64_t expectedLength;
    CFStringRef textEncoding;
    CFIndex statusCode;
    CFStringRef httpVersion;
    CFDictionaryRef headerFields;
    Boolean isHTTPResponse;

    OSSpinLock parsedHeadersLock;
    ParsedHeaders* parsedHeaders;
} _CFURLResponse;

typedef const struct _CFURLResponse* CFURLResponseRef;

你可以通过以下代码从NSURLResponse中获取到这个结构体

SEL selector = NSSelectorFromString(@"_CFURLResponse");
CFTypeRef response = CFBridgingRetain([response performSelector:selector]);
CFShow(response);

拿到CFURLResponseRef,又要怎么从它获取到httpVersion呢?无意间又发现下面这个函数可以将CFURLResponseRef转化为CFHTTPMessageRef

CFHTTPMessageRef CFURLResponseGetHTTPResponse(CFURLResponseRef);

而下面这个函数又可以从CFHTTPMessageRef中获取到我们想要的HttpVersion

CFStringRef CFHTTPMessageCopyVersion(CFHTTPMessageRef message);

四、示例代码

#import <dlfcn.h>
#import "NSURLResponse+Help.h"

@implementation NSURLResponse (Help)

typedef CFHTTPMessageRef (*MYURLResponseGetHTTPResponse)(CFURLRef response);

- (NSString *)getHTTPVersion {
    NSURLResponse *response = self;
    NSString *version;

    // 获取CFURLResponseGetHTTPResponse的函数实现
    NSString *funName = @"CFURLResponseGetHTTPResponse";
    MYURLResponseGetHTTPResponse originURLResponseGetHTTPResponse =
        dlsym(RTLD_DEFAULT, [funName UTF8String]);

    SEL theSelector = NSSelectorFromString(@"_CFURLResponse");
    if ([response respondsToSelector:theSelector] &&
        NULL != originURLResponseGetHTTPResponse) {
        // 获取NSURLResponse的_CFURLResponse
        CFTypeRef cfResponse = CFBridgingRetain([response performSelector:theSelector]);
        if (NULL != cfResponse) {
            // 将CFURLResponseRef转化为CFHTTPMessageRef
            CFHTTPMessageRef message = originURLResponseGetHTTPResponse(cfResponse);
            // 获取http协议版本
            CFStringRef cfVersion = CFHTTPMessageCopyVersion(message);
            if (NULL != cfVersion) {
                version = (__bridge NSString *)cfVersion;
                CFRelease(cfVersion);
            }
            CFRelease(cfResponse);
        }
    }

    // 获取失败的话则设置一个默认值
    if (nil == version || 0 == version.length) {
        version = @"HTTP/1.1";
    }

    return version;
}

@end

五、最后的话

  1. _CFURLResponse和CFURLResponseGetHTTPResponse都是苹果没有公开的,使用时需要特殊处理下,以防上架时被苹果判定为使用了Private API而被拒。至于怎么处理就不细说了,可以采用拼接字符串或者对其进行加解密的方式。
  2. 之前做HTTP2的时候,不知道在客户端要怎样获取请求使用的HTTP协议版本,以判断是否协商使用了HTTP2协议,都是通过在服务端查看日志来判断的。现在好了,直接通过NSURLResponse就可以获取到了。
  3. 说来惭愧,工作多年了还是头一回写博客。哪里写得不好或者有什么错误,还请大家多多包涵并给予指正。
iOS开发
Web note ad 1